每日一题 | LeetCode 1 两数之和

2022 年 6 月 06 日
本文字数：1072 字
阅读完需：约 4 分钟

题意

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

创建一个哈希表。
遍历 nums 数组，对于当前元素 nums[i]，查询哈希表中是否存在 target - nums[i]。
若存在，返回下标，如不存在，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;    }}

复制代码

发布于: 刚刚阅读数: 5

原文链接:【http://xie.infoq.cn/article/426a954832ac52d4dcaeee5b0】。文章转载请联系作者。

武师叔

关注

每天丰富自己，去过自己想要的生活！ 2022.04.28 加入

一个喜欢最新技术，研发的人工智能专业的大二学生，用自己的代码做一些有意义的事情！目前大二结束有去大厂研发岗实习的计划，每天丰富自己的技术，去过自己想要的实习生活。

发布

暂无评论

创作场景

每日一题 | LeetCode 1 两数之和

题意

示例

题目解析

代码示范

Python 代码

Java 代码

武师叔

评论