1. 程式人生 > >LeetCode刷題之求兩數之和

LeetCode刷題之求兩數之和

題目:
給定一個整數陣列和一個目標值,找出陣列中和為目標值的兩個數。

你可以假設每個輸入只對應一種答案,且同樣的元素不能被重複利用。
示例:
給定 nums = [2, 7, 11, 15], target = 9

因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

解決思路:
1.遍歷兩次列表,然後判斷遍歷得到的數之和是否等於目標值,是:返回,否:繼續遍歷
2.遍歷一次列表,利用減法,判斷另一個數是否在列表中,是:返回,否:繼續。
二者對比,第一種思路遍歷兩次,時間複雜度比第二種高,故採用第二種解法。

l = []
def two_sum(nums, target):
    n = len(nums)
    for i in range(n):
        a = target - nums[i]
        if a in nums[i + 1::]:
            b = nums.index(a)
            l.append(i)
            l.append(b)
            return l
        else:
            continue


nums = [2, 7, 11, 19]
ret = two_sum(nums, 9)
print(ret)