給定一個整數數組和一個目標值,找出數組中和為目標值的兩個數。你可以假設每個輸入只對應一種答案,且同樣的元素不能被重覆利用。 示例: 給定 nums = [2, 7, 11, 15], target = 9 因為 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1] 代碼 ...
給定一個整數數組和一個目標值,找出數組中和為目標值的兩個數。你可以假設每個輸入只對應一種答案,且同樣的元素不能被重覆利用。
示例:
給定 nums = [2, 7, 11, 15], target = 9
因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
代碼實現:
def twoSum(self, nums, target):
nums_bak = nums.copy()
nums.sort()
i = 0
j = 0
for k in range(0, (len(nums) - 1)):
if nums[k] + nums[k + 1] >= target:
i = k
j = k + 1
break
while i >= 0 and j < len(nums):
if nums[i] + nums[j] < target:
j += 1
elif nums[i] + nums[j] > target:
i -= 1
else:
if nums[i] == nums[j]:
return [nums_bak.index(nums[i]), nums_bak.index(nums[j], i + 1)]
else:
return [nums_bak.index(nums[i]), nums_bak.index(nums[j])]