LeetCode1、两数之和

218 阅读2分钟

LeetCode 系列记录我学习算法的过程。

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第1天,点击查看活动详情

题目

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

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

示例:

输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。

输入: nums = [3,2,4], target = 6
输出: [1,2]

输入: nums = [3,3], target = 6
输出: [0,1]

提示

  • 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++) {
            if (nums[i] + nums[j] === target) {
                return [i, j]
            }
        }
    }
};

提交代码,完美通过😂

image.png

优化

看到执行用时和内存消耗都远不如别人,就去看了下别人的题解

果然,我这种方法被称为暴力枚举,是最简单也最没技术含量的😂

然后看了下别人的解题思路:

通过 Map 哈希表来解决,第一层 for 循环是少不了的,遍历到当前项 a 时,用 target 减去 a 得到的 b 去哈希表里找,找到了就返回哈希表里索引 b 对应的值和当前的索引,没找到就把 akey,当前索引为值,存入哈希表里,然后遍历下一项,下面是优化后的代码实现:

/**
 * @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 ++) {
        if (map.has(target - nums[i])) {
            return [map.get(target - nums[i]), i]
        } else {
            map.set(nums[i], i)
        }
    }
};

image.png 速度确实比我上一次的快了不少,之前还真没想过 Map 还能这样用,学到了