每日一题 | LeetCode 1 两数之和

每日一题 | LeetCode 1 两数之和

题意

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

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

示例

示例 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]

题目解析

难度：简单

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

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

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

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

两数之和的求解可以分为以下 3 步：

1. 创建一个哈希表。
2. 遍历 nums 数组，对于当前元素 nums[i]，查询哈希表中是否存在 target - nums[i]。
3. 若存在，返回下标，如不存在，nums[i] 存入哈希表，继续第 2 步。

图解
以 nums = [2,7,11,15]，target = 9 为例。

首先初始化一个哈希表，key 存储 nums[i]，value 存储 i。

代码示范

Python代码

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;
    }
}