Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution.
執行個體:
輸入: numbers={2, 7, 11, 15}, target=9
輸出: (0,1)
題目分析:
給定一個整數數組,任意輸入一個數,在數組中尋找並返回能夠相加得到這個數的兩個數的位置。不能排序,否則會打亂原始數組內的位置,題目要求就是找到原始位置。
方法一:
思路:用雙重迴圈,得到任意組合相加的和,判斷是否與輸入的數相等。
缺點:時間複雜度過高為O(n²),運行速度大打折扣。而且提交時因為時間複雜度會提示不合格 優點:最簡單的思路 代碼:(此代碼不能成功提交,原因↑)
class Solution: def twoSum(self, nums, target): for i in range(len(nums)): a=nums[i] for j in range(len(nums)): if target-a==nums[j]: return i,jt=Solution()arr=[1,2,3,6]jg=t.twoSum(arr,4)print(jg)
運行結果:(0,2)
方法二: 思路:要解決時間複雜度的問題,就要簡化迴圈,只使用一重迴圈,在參考了別人的代碼後,探索方法就是建立一個空字典,通過迴圈往進添加元素的同時尋找已添加的元素中是否有合格結果。 優點:時間複雜度為O(n),空間複雜度為O(n)。 缺點:難理解。 代碼:(成功提交)
class Solution: def twoSum(self, nums, target): dict={} for i in range(len(nums)): x=nums[i] if target-x in dict: return (dict[target-x],i) dict[x]=i #此時確定了字典內的內容為dict{x:i}注意這裡的KEY(其實就是順序123)放在了i的位置,x的位置放的是數組中元素。這樣上一行的dict[target-x]返回的才是位置t=Solution()arr=[3,2,3]jg=t.twoSum(arr,6)print(jg)
總結:用方法二才可以通過。我最開始的思路太簡單,沒有考慮到時間複雜度的問題。以後在做題過程中一定要考慮運行速度的問題。網上還有好多種解法,冒泡排序,雜湊函數之類的,有空要去試著理解一下,運行速度應該比我這種蠢方法好。但我覺得還是我這兩種方法好理解。而且作為個初學者應該也還可以吧。