Skip to main content

两数之和

给定一个整数数组 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]

我的菜鸡解法​

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
var twoSum = function (nums, target) {
  let store = {};
  let res = [];
  nums.forEach((item, index) => {
    if (store[item] === undefined) {
      store[item] = {
        value: item,
        pos: index,
      };
    }
    if (
      store[target - item] !== undefined &&
      store[target - item].value !== undefined
    ) {
      if (index !== store[target - item].pos) {
        res = [store[target - item].pos, index];
      }
    }
  });
  return res;
};

缺点

  1. forEach不可中断,一定会遍历完, 损失性能
  2. 不需要存储额外的 value,只需要存下标,占用内存

优质解法​

function twoSum(nums: number[], target: number): number[] {
  let map = {};
  for (let i = 0; i < nums.length; ++i) {
    const rest = target - nums[i];
    if (map[rest] !== undefined) {
      // 注意这里map里面的是下标比较小的值。
      return [map[rest], i];
    }
    map[nums[i]] = i;
  }
  return [];
}