[路飞]_js算法:leetcode 1-两数之和

48 阅读1分钟

leetcode 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]

提示:

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

进阶: 你可以想出一个时间复杂度小于 O(n2) 的算法吗?

思路: 根据提示,时间复杂度我们可以想到二分算法。生成一个排序以后的下标数组,然后用二分算法来查找对应的值

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
var twoSum = function(nums, target) {
    let indexArr=new Array(nums.length)
    for(let i=0;i<indexArr.length;i++){
        indexArr[i]=i
    }
    indexArr.sort((i,j)=>nums[i]-nums[j]);
    let res=[-1,-1]
    for(let i=0;i<indexArr.length;i++){
        let val=nums[indexArr[i]];
        let j=findTarget(nums,indexArr,i+1,target-val);
        res[0]=indexArr[i];
        if(j>=0){
            res[1]=indexArr[j];
            return res

        }
    }
};
function findTarget(nums,indexArr,head,target){
    let heads=head,tail=indexArr.length-1,mid;
    while(heads<=tail){
        mid=heads+Math.floor((tail-heads)/2);
        if(nums[indexArr[mid]]===target)return mid;
        else if(nums[indexArr[mid]]<target)heads=mid+1
        else{
            tail=mid-1
        }
    }
    return -1
}