給定一個整數數組 nums 和一個目標值 target,請你在該數組中找出和爲目標值的那 兩個 整數,並返回他們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,數組中同一個元素不能使用兩遍。
示例:
給定 nums = [2, 7, 11, 15], target = 9
因爲 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路:依稀記得第一次做leetcode的時候,看到這個題覺得哇太簡單了,兩層for循環就行了,然後結果就是超出時間限制。第一道題呀,簡直就被秀了一臉,完了之後稍微改良了一下可以通過,不過這個運行時間是真的太長了,當時看hash還不理解這種解法巧妙在何處,現在終於終於理解了,做個記錄。先說採用hash的方法:key是元素的值,value是元素的索引。然後遍歷這個數組,在hash表中去查找是否有這個元素,然後返回索引。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
'''
#暴力改良版,時間非常的長
for i in range(len(nums)):
for j in range(i+1,len(nums)):
if nums[j] == target-nums[i]:
return [i,j]
'''
#哈希列表;初始化啥也沒有就是一個空的哈希表
has = dict()
res = 0
#遍歷整個數組,然後不斷的更新哈希表,如果res的結果在哈希表中的話,那麼返回下標索引就可
for i in range(len(nums)):
res = target-nums[i]
if res in has.keys():
return [has[res],i]
has.update({nums[i]:i})
總結:這個題還是用空間換時間,採用哈希表就可以一下將O(n²)的複雜度降低到O(n).