题目:给定一个整数数组 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) 的算法吗?
题解:
直接想进阶吧!暴力解法就是双重循环找目标值的解。
算法一:
把nums
中的id
与数据对应起来,存放到新的容器中,按照数据的大小对容器进行排序,得到数据大小排序对应的id号,然后循环遍历id,以头+尾的方式计算id
对应在nums
中的两数之和sum
,如果sum
<target
,那么head
++;反之tail
–。循环终止的条件是两数之和sum=target。
算法二:
构造一个哈希表,存放对应的数据和他的下标id
,然后遍历这个哈希表,在循环中的每一个num
,都找对应的target
-num
是否存在于这个哈希表中,如果存在,那么当即跳出循环;否则继续找。
代码:
bool cmp(pair<int,int> a,pair<int,int> b)
{
return a.first<b.first;
}
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ans;
vector<int> sorted_num=nums;
vector<pair<int,int>> sorted_pair;
vector<int> sorted_id;
for(auto i=0;i<sorted_num.size();i++)//构造id与数据的对应关系
{
sorted_pair.push_back(make_pair(nums[i],i));
}
sort(sorted_pair.begin(),sorted_pair.end(),cmp);//按照数据大小排序
for(auto i=0;i<sorted_num.size();i++)//存放排序后的id
{
sorted_id.push_back(sorted_pair[i].second);
}
int head=0,tail=nums.size()-1,sum=nums[sorted_id[head]]+nums[sorted_id[tail]];
while(sum!=target)//循环查找
{
if(sum<target)
{
head++;
}
else if(sum>target)
{
tail--;
}
sum=nums[sorted_id[head]]+nums[sorted_id[tail]];
}
ans.push_back(sorted_id[head]);
ans.push_back(sorted_id[tail]);
return ans;
}
};