使用 Python 編寫的程式,該程式包括一個 twoSum 方法來找出數組中兩個整數相加等於目標值的索引位置,並回傳這些索引組成的數組。如果找不到,則回傳 None。同時,主程式負責呼叫 twoSum 方法並顯示結果。
python
複製程式碼
def twoSum(dataArray, targetValue):
for i in range(len(dataArray)):
for j in range(i + 1, len(dataArray)):
if dataArray[i] + dataArray[j] == targetValue:
return [i, j]
return None
def main():
# 測試一
dataArray = [3, 11, 14, 6]
targetValue = 9
result = twoSum(dataArray, targetValue)
print(f"測試一結果: {result}") # 應顯示 [0, 3]
# 測試二
dataArray = [3, 11, 14, 6]
targetValue = 18
result = twoSum(dataArray, targetValue)
print(f"測試二結果: {result}") # 應顯示 None
if __name__ == "__main__":
main()
說明
twoSum 函式:此函式接受兩個參數,分別是數組 dataArray 和目標值 targetValue。它使用雙重迴圈來檢查數組中的每一對元素,並判斷它們的和是否等於目標值。如果找到符合條件的一對元素,回傳它們的索引。如果遍歷整個數組後仍未找到,則回傳 None。
main 函式:這是主程式,包含兩組測試資料。它呼叫 twoSum 函式並顯示結果。