每日一题 | 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 步。
图解
以 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;
}
}