【LeetCode 热题 HOT 100-001】两数之和（python）

题集链接：https://leetcode.cn/problem-list/2cktkvj/

题目链接：https://leetcode.cn/problems/two-sum/?favorite=2cktkvj

一、题目

二、代码

解法1：暴力求法

解法2：使用字典

解法3：哈希表

三、总结

---

一、题目

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

提示：

• 2 <= nums.length <= 104
• -109 <= nums[i] <= 109
• -109 <= target <= 109
• 只会存在一个有效答案

进阶：你可以想出一个时间复杂度小于 O() 的算法吗？

二、代码

解法1：暴力求法

直接使用两个for循环

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        for i in range(len(nums)):
            for j in range(i+1, len(nums)):
                if(nums[i]+nums[j] ==target):
                    return [i,j]

解法2：使用字典

以 nums = [2,7,11,15], target = 9 为例，构造字典

num_dict [2] = 0

num_dict [7] = 1

num_dict [11] = 2

num_dict [15] = 3

tar = 9-2 = 7

if 7 in [2,7,11,15]

return [0,1]

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        num_dict = {}
        for i, num in enumerate(nums):
            num_dict[num] = i    # num_dict:[2:0,7:1,11:2,15:3]
        for j in range(len(nums)):
            tar = target - nums[j]
            if (tar in num_dict.keys()) and (num_dict[tar]!=j):    # num_dict.keys()获取所有的键
                return [j, num_dict[tar]]

解法3：哈希表

以 nums = [3, 2, 4], target =6 为例，构造哈希表

hashtable [3] = 0

hashtable [2] = 1

target - num = 6-4 = 2

return [hashtable[target - num], i] = [1 , 2]

解法1、2，从头向后找；解法3从后往前找

对于每一个 x，我们首先查询哈希表中是否存在 target - x，然后将 x 插入到哈希表中，即可保证不会让 x 和自己匹配。

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

三、总结

enumerate() 函数：将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列，同时列出数据和数据下标 https://zhuanlan.zhihu.com/p/92544989

example = 'abcd'
for i,j in enumerate(example):print(i,j)

打印结果为：
0 a
1 b
2 c
3 d