看到了就一块做了,两个题的要求差不多,条件不同:
给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n,不是以 0 开头。
样例
给出 numbers = [2, 7, 11, 15], target = 9, 返回 [1, 2].
两个数的条件不同,56给出的数组是乱序的,608给的数组是顺序的。
暴力查找
对于每一个数,在数组中查找和其和为目标数的数,注意两个数的索引不能相同,这样的时间复杂度是O(n^2),谁都能想到,不推荐。
排序加双指针
608
这个思路就比较好了,对于608给出的数据,直接排序加双指针就ok了,记得最后给索引加1。
vector<int> twoSum(vector<int> &nums, int target) {
auto beg=nums.begin();
auto end=nums.end()-1;
vector<int> res;
int s=0;
//两头
while(beg<end)
{
if(*beg+*end==target) //等于的话就刚刚好
{res.push_back(beg-nums.begin()+1);
res.push_back(end-nums.begin()+1);
return res;
}
if(*beg+*end<target) //小于的话就增加第一个
{
beg++;
}
else //大于的话就减小后一个
{
end--;
}
}
// write your code here
}
56
这个倒是可以排序,但是记得要同时把索引也保存起来,因为最后要求返回的是索引,可以把值和索引做成一个pair放入vector里面,然后根据值来排序,这里的排序规则要自己写,sort函数的第三个参数是支持的,然后再双指针就可以了。
static bool MyComp(pair<int, int> a, pair<int, int> b)
{
return a.first < b.first;
}
vector<int> twoSum(vector<int> &numbers, int target) {
vector<int> res;
vector<pair<int, int>> nums;
for (size_t i = 0; i<numbers.size(); i++)
{
nums.push_back(make_pair(numbers[i], i)); //把key和value分别存入map
}
sort(nums.begin(), nums.end(),MyComp); //自定义排序规则
auto beg = nums.begin();
auto end = nums.end() - 1;
while (beg != nums.end())
{
if(beg->first+end->first==target)
{
res.push_back(beg->second + 1);
res.push_back(end->second + 1);
break;
}
else if (beg->first + end->first < target)
beg++;
else if (beg->first + end->first > target)
end--;
}
sort(res.begin(),res.end());
return res;
// write your code here
}
既然用pair来做可以,那么干脆用map来做就行了,因为map是按照key进行自动排序的,不用自己排序,要注意需要用multimap来做,不要去重,用insert添加进去。
但是map就不能用双指针了,map的迭代器不支持减操作,但是map自己的find函数还是可以用的,按照key查找,返回对应的迭代器,如果不存在则返回尾后。
这样说起来vector也是可以用find的,但是为了体会这个过程,还是不用了的好。程序如下:注意map查找时,可能返回当前遍历的元素,要排除掉。
vector<int> twoSum(vector<int> &numbers, int target) {
vector<int> res;
multimap<int,int> nums;
for (size_t i = 0; i<numbers.size(); i++)
{
nums.insert(make_pair(numbers[i],i)); //把key和value分别存入map
}
auto beg = nums.begin();
while (beg != nums.end())
{
auto f = nums.find(target-beg->first);
if (f != nums.end()&&f!=beg) //不能等于自己,要不两个索引是一样的了。
{
res.push_back(beg->second + 1);
res.push_back(f->second + 1);
break;
}
beg++;
}
sort(res.begin(), res.end());
return res;
// write your code here
}