leetcode-两数之和

目录

• 1.题目描述
• 2. 解题思路
  • a. 暴力法
  • b. 两遍哈希
  • c. 一遍哈希

 

---

1.题目描述

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那 两个 整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

示例

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

2. 解题思路

a. 暴力法

枚举数组中每一个元素$x$,判断是否存在 $target-x$.

    def twoSum1(self, nums: List[int], target: int) -> List[int]:
        n = len(nums)
        for i in range(n-1):
            for j in range(i+1,n):
                if nums[i]+nums[j]==target:
                    return [i,j]
        return []
 

b. 两遍哈希

第一次遍历，将元素的值和索引加入哈希表中（字典），
第二次遍历，判断$target-x$是否存在与哈希表中。
注意：目标元素不能是x本身。

    def twoSum2(self, nums: List[int], target: int) -> List[int]:
        n = len(nums)
        hash = {}
 
        for i in range(n):
            hash[nums[i]]=i
        
        for j in range(n):
            complement = target - nums[j]
            if complement in hash and j!=hash[complement]:
                return [j,hash[complement]]
 
        return []

c. 一遍哈希

针对方法二可以优化，将两次遍历合并到一次。对于每一个x, 首先判断哈希表中是否存在$target-x$, 然后将x插入到哈希表中。

    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 []