Leetcode 两数之和

178 阅读2分钟

「这是我参与2022首次更文挑战的第1天,活动详情查看:2022首次更文挑战」。

题目🌵

📝Leetcode 两数之和

✏️leetcode-cn.com/problems/tw…


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

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

你可以按任意顺序返回答案。

示例 1:

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

解题思路💡

暴力解法

  • 首先想到的是直接莽,直接对所有的数进行循环
  • 当找到nums[i]+nums[j]==target时返回当前的索引值
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] == target - nums[j]) {
                return [i, j]
            }
        }
    }
    return []
}

image-20220119145221850

进一步优化

从图上所示,我们执行的时间靠后,说明程序执行消耗了很多时间,我们可以做进一步的优化。

🔧问题:当与第一个数做匹配时,需要遍历一次数组,消耗很多的时间

💡思路:建立一个HashMap,记录当前的差值和对应的索引号,避免了重复遍历前面的元素。

Hash表

  • 先判断Hash表中是否存在target-nums[i]
  • 如果存在,返回当前的索引值和map.get(target-nums[i])
  • 如果不存在,则以num[i]作为key,以索引地址作为value
var twoSum = function (nums, target) {
    const map = new Map()
    for (let i = 0, len = nums.length; i < len; i++) {
        let match = target - nums[i]
        if (map.has(match)) {
            return [map.get(match), i]
        } else {
            map.set(nums[i], i)
        }
    }
    return []
}

也可以直接使用in操作符

var twoSum = function (nums, target) {
  let obj = {}
  for (let i = 0; i < nums.length; i++) {
    let num = nums[i]
    if (num in obj) {
      return [obj[num], i]
    } else {
      obj[target - num] = i
    }
  }
  return []
}

image-20220119145136564

可以看出经过优化后,程序在性能上得到了优化。