LeetCode刷題之求兩數之和
阿新 • • 發佈:2018-11-19
題目:
給定一個整數陣列和一個目標值,找出陣列中和為目標值的兩個數。
你可以假設每個輸入只對應一種答案,且同樣的元素不能被重複利用。
示例:
給定 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)