leetcode 刷题 —— twoSum

289 阅读1分钟

题目

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]

解题思路

遍历 nums,用一个hashmap 存储已经访问过的数字和对应的下标,若 hashmap 中存在一个键与 target - nums[i] 相等,说明找到了与该数字相加等于 target 的数字,返回 hashmap 中对应的键值和该数字的下标。

js demo

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
var twoSum = function (nums, target) {
  const visited = {} // 记录出现的数字, 空间复杂度N

  for (let index = 0; index < nums.length; index++) {
    const element = nums[index];

    if (visited[target - element]) {
      return [visited[target - element] - 1, index]
    }
    // 这里为什么不直接等于 index 呢?因为 index 会有为 0 的情况,这个时候,上述写法就有问题了,因为 0 为 false
    visited[element] = index + 1;
  }
  return [];
};

console.log(twoSum([2, 7, 11, 15], 22))