一、题目描述
给定一个整数数组 nums 和一个整数目标值 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]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
提示:
- 2 <= nums.length <= 10^3
- 10^9 <= nums[i] <= 10^9
- 10^9 <= target <= 10^9
- 只会存在一个有效答案
二、思路分析
这道题我刚看到时的基本思路:
两层循环来遍历同一个数组;
第一层循环遍历的值记为 value1,
第二层循环时遍历的值记为 value2;
若 value1+value2 = 目标值,
那么 value1 和 value2 对应的数组下标就是我们想要的答案。
!!!注意,两层循环很多情况下都意味着 O(n^2) 的复杂度,这个复杂度非常容易导致算法超时,必须得优化一下。
试着用空间换时间,把它优化成一层循环:
我们可以在遍历数组的过程中,增加一个 Map 来记录已经遍历过的数字及其对应的索引值。
然后每遍历到一个新数字的时候,都回到 Map 里去查询 targetNum 与该数的差值是否已经在前面的数字中出现过了。
若出现过,那么答案已然显现,我们就不必再往下走了。
三、AC 代码
/**
* @param {number[]} nums
* @param {number} target
* @return {number[]}
*/
var twoSum = function(nums, target) {
let map = new Map();
// 遍历数组
for(let i = 0;i<nums.length;i++){
let key = target - nums[i];
// 判断当前值对应的 target 差值是否存在(是否已遍历过)
if(map.has(key)){
// 若有对应差值,那么答案get!
return [map.get(key),i];
}
// 若没有对应差值,则记录当前值
map.set(nums[i],i);
}
};
四、总结
- 当发现自己的代码里有两层循环时,先反思一下,能不能用空间换时间,把它优化成一层循环。
- 几乎所有的求和问题,都可以转化为求差问题。 这道题就是一个典型的例子,通过把求和问题转化为求差问题,事情会变得更加简单。