一、题目
给定一个整数数组 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) 的算法吗?
二、思路
先前做过三数之和,我直接套用思路了(虽然感觉简单题二重for应该也能过)。
思路算是双指针吧,排序之后左指针指向小的数字,右指针指向大的数字,相加小于target就左指针右移,大于target则右指针左移。
记得用pair<int,int>来存储一下原先的下标即可。
三、代码
cpp
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
vector<pair<int,int>> arr;
bool found=false;
for(int i=0;i<nums.size();i++){
arr.push_back({nums[i],i});
}
sort(arr.begin(),arr.end());
for(int i=0;i<arr.size();i++){
int left=i;
int right=arr.size()-1;
while(left<right){
int sum=arr[left].first+arr[right].first;
if(sum<target){
left++;
}
else if(sum>target){
right--;
}
else{
res.push_back(arr[left].second);
res.push_back(arr[right].second);
found=true;
break;
}
}
if(found) break;
}
return res;
}
};