题目描述
来源:力扣(LeetCode)
题目链接:https://leetcode-cn.com/problems/two-sum
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 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]。
tip
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
分析解题
拿到题目,马上就会想到使用双重遍历把元素全部比较一遍再选出正确结果,代码如下:
/**
* @param {number[]} nums
* @param {number} target
* @return {number[]}
*/
var twoSum = function (nums, target) {
// 双重遍历全部元素
for (let i = 0; i < nums.length; i++) {
for (let j = i + 1; j < nums.length; j++) {
// 两两相加判断是否与target相等
if (nums[i] + nums[j] === target) {
// 相等则返回索引
return [i, j]
}
}
}
return []
};
这种方法简洁明了,但是也预想到了运行时长久的结果,120ms仅仅击败了12%的提交记录,这也难怪,毕竟使用双重遍历时间复杂度为O(n2)
想要使时间复杂度低于O(n2),就必须打破双重遍历,有什么办法能使用一次遍历就把所有的元素相加一次呢?
其实可以在遍历时把之前的数据存储起来,同时还要记录索引,最好的方法就是使用Map对象来存储数据,简化后代码如下
/**
* @param {number[]} nums
* @param {number} target
* @return {number[]}
*/
const twoSum = (nums, target) => {
// 创建Map对象
const map = new Map()
for (let i = 0; i < nums.length; i++) {
// 判断map中是否存在与当前遍历值相加等于target的元素
if (map.has(target - nums[i])) {
// 存在则直接返回当前索引和匹配值的索引
return [i, map.get(target - nums[i])]
}
// 没有则把当前遍历值和索引存储到map中
map.set(nums[i], i)
}
return []
}
这样时间复杂度就变为了O(n),不过空间复杂度稍稍提高,运行速度始终保持在70ms左右,比起之前的双重遍历也算是有很大的提升了
补充js的Map对象及其用法:https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Map