LeetCode -- 1:两数之和(简单)

28 阅读2分钟

1. 两数之和

题目描述

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

常规解法

关键思路 - 暴力匹配

  • 最暴力的解法:用两层遍历,遍历所有两个数相加的情况。

JS代码实现

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
const twoSum = (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];
      }
    }
  }
};

复杂度分析

  • 时间复杂度:O(n²)
  • 空间复杂度:O(1)

优化解法

上述实现代码的时间复杂度:O(n²),有没有更优解呢,让时间复杂度小于O(n²)?

关键思路 - map映射

题目描述是在该数组中找出 和 为目标值 target 的那 两个 整数,即:

num[i]+num[j]=targetnum[i] + num [j] = target

那我们也可以换一个思路:

x=targetnum[i] x = target - num[i]

这样,我们就只需要查找 x 是不是存在数组中了。

JS代码实现

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
const twoSum = (nums, target) => {
  const numMap = {};
  for (let i = 0; i < nums.length; i++) {
    const num = nums[i];
    let x = target - num;
    // 此处判断要注意,不能用if(numMap[x]),因为x可能为0
    if (numMap[x] !== undefined) {
      return [numMap[x], i];
    }
    numMap[num] = i;
  }
  return [];
};

复杂度分析

  • 时间复杂度:O(n)
  • 空间复杂度:O(n)

结语

本文主要介绍了LeetCode第1题两数之和的两种解法,你还知道这题的其他解法吗?欢迎在评论区留言分享!