【刷题日记】两数之和

两数之和这个题在leetcode里面算是词典里面abandon'一样的存在了

 

力扣

 两个for循环的话实际上是最基础的，这里就不赘述了，这里主要是想要得到一种时间复杂度小于o(n^2)的算法，那么第一个想到的是哈希表，那么开动吧

第一个想到的就是先把所有的数都先存在哈希表中，存完之后再遍历哈希表进行计算，由于存在重复数字的可能，使用列表进行存储对应的idx，若target-当前key存在，则说明列表中有与当前key相加为target的数字，那么就完成了计算。这里的时间复杂度是o(2N)

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dict_nums = {}
        for idx,key in enumerate(nums):
            if key not in dict_nums:
                dict_nums[key] = [idx]
            else:
                dict_nums[key].append(idx)
        for key,value in dict_nums.items():
            idx_1 = dict_nums[key].pop(0)
            idx_list = dict_nums.get(target-key)
            if idx_list is not None and idx_list != []:
                return [idx_1, idx_list[0]]

第二种方法是官方提供的，最开始还没看懂，看懂了之后觉得很是精妙，

首先定义一个空的哈希表，因为实际上列表中如果存在两数相加的可能，即a+b=target，

那么第一次出现的a我们可以先加在字典中，

在b出现的时候我们自然可以通过相减的方式找到已经加在字典中的a，

所以我写的先遍历创建字典实际上是有点画蛇添足了，这里的时间复杂度是O(N)，学到了

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        hashtable = dict()
        for i, num in enumerate(nums):
            if target - num in hashtable:
                return [hashtable[target - num], i]
            hashtable[nums[i]] = i
        return []

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/two-sum/solution/liang-shu-zhi-he-by-leetcode-solution/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。