每日一题 | 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 步:
- 创建一个哈希表。
- 遍历 nums 数组,对于当前元素 nums[i],查询哈希表中是否存在
target - nums[i]
。 - 若存在,返回下标,如不存在,nums[i] 存入哈希表,继续第 2 步。
代码示范
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; } }