每日算法【数组】 - 两数之和

85 阅读2分钟

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

好好学习,天天刷题~~

今日份刷题两数之和记录如下:

题目描述

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

提示:

2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案

暴力破解

最简单的方法就是暴力破解啦,需要两次遍历,寻找target - x的值,一旦找到,则返回对应的位置坐标即可,非常简单,但是时间复杂度为O(N^2),不是很理想~

var twoSum = function(nums, target) {
  let left=0;
  while(left<nums.length){
    for(var i=(left+1);i<nums.length;i++){
    if(nums[left]+nums[i]==target){
      var arr = [left,i];
      return arr;
    }
   }
   left++;
  }  
};

不出意外,提交结果如下: image.png

巧用map法

这道题稍微琢磨一下,发现可以用map解法,map是以key和value键值对存在的,也提供了has方法去判断map中是否存在key值,还有get方法返回map中的value值,set方法给map对象中的键设置值。 那么我们可以这么做:

  • 创建一个map实例
  • 遍历数组,并为map新增当前数值的键和值
  • 查找数组中是否存在target减当前数组值的值x,如果存在则直接返回存在x的map的键 具体代码如下:
var twoSum = function(nums, target) {
  var arr= new Map();
  for(let i=0;i<nums.length;i++){
    var x = target - nums[i];
    if(arr.has(x)){
      return [arr.get(x),i];
    }
    arr.set(nums[i],i);
  }
};

提交结果如下:

image.png