《Leetcode》1、兩數之和

給定一個整數數組 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).

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章