两数之和

80 阅读2分钟

题目描述

给定一个整数数组 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]
提示
  • 2 <= nums.length <= 104
  • -109 <= nums[i] <= 109
  • -109 <= target <= 109
  • 只会存在一个有效答案
进阶:

你可以想出一个时间复杂度小于 O(n2) 的算法吗?

思路分析:

我一开始想的是直接两个for循环去遍历数组,两次遍历相加的结果等于target就可以直接return数组下标。

代码
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
var twoSum = function (nums, target) {
    for (let i = 0; i < nums.length - 1; i++) {
        for (let j = i + 1; j < nums.length; j++) {
            if (nums[i] + nums[j] === target) {
                return [i, j]
            }
        }
    }
};
缺陷

这种思路非常消耗内存和时间,可以看到测试用例执行结果:

image.png

第二种方法:

思路:是利用map来查找和存储target-i的差值及其下标,在map里如果没有就用map.set()方法存储nums[i]和i,用map.has(差值)来查找是否有这个值,有的话就返回。

代码:
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
var twoSum = function (nums, target) {
    const map = new Map
    for (let i = 0; i < nums.length; i++) {
        let diffValue = target - nums[i]
        if (map.has(diffValue)) {           //得到的是map里的key
            return [map.get(diffValue), i]
        } else {
            map.set(nums[i], i)
        }
    }
    return []
};

这种思路需要掌握map结构和它的方法,其所消耗的时间和内存都大大减少了。

image.png