每日一题| 1. 两数之和

每日一题| 1. 两数之和

题目

给定一个整数数组 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]

思路

水题，难度简单。

这道题如果不考虑时间复杂度的话，最简单粗暴的方式就是暴力破解，枚举 nums 中的 x，再找数组中是否存在 target -x。

这种双重循环的时间复杂度是 O(n²)，这也太慢了，可以想一个时间复杂度小于 O(n) 的算法嘛？

答案肯定是可以的，用哈希可以高效解决。

为啥这么笃定呢？

就源于一句“找数组中是否存在 target -x”，在一堆数中找一个数用哈希没跑儿，毕竟作为查找中的秒男，哈希查找的时间复杂度是 O(1)。

哈希表

希函数就是将key和value值对应起来的函数。在不同语言中，哈希函数的实现方式是不一样的，在python语言中，哈希表对应的集合叫做字典（dict）。

解题流程

• 创建一个哈希表。

• 遍历 nums 数组，对于当前元素 nums[i]，查询哈希表中是否存在 target - nums[i]。

• 若存在，返回下标，如不存在，nums[i] 存入哈希表，继续第 2 步。
  

代码展示

Python3

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        # 初始化哈希表
        hash = dict()
        # 遍历数组 nums
        for i in range(len(nums)):
            # 若 target - nums[i] 在哈希表中
            if target - nums[i] in hash:
                # 返回结果
                return [hash[target - nums[i]], i]
            # 若 target - nums[i] 不在哈希表中，将 nums[i] 存入哈希表
            hash[nums[i]] = i

        return []

Java

class Solution {
    public int[] twoSum(int[] nums, int target) {
        if (Objects.isNull(nums) || nums.length == 0) return null;
        //初始化哈希表 
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int curNum = target - nums[i];
            //是否存在 target - nums[i]的值
            if (map.containsKey(curNum)) {
                return new int[]{map.get(curNum), i};
            }
            //不存在 , 则进行存储
            map.put(nums[i], i);
        }
        return null;
    }
}