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
}