两数之和
描述
笔记
数据
评测
给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n,不是以 0 开头。
注意事项
你可以假设只有一组答案。
您在真实的面试中是否遇到过这个题? Yes
样例
给出 numbers = [2, 7, 11, 15], target = 9, 返回 [1, 2].
标签
最初代码:
class Solution {
public:
/*
* @param numbers : An array of Integer
* @param target : target = numbers[index1] + numbers[index2]
* @return : [index1+1, index2+1] (index1 < index2)
*/
vector<int> twoSum(vector<int> &nums, int target) {
// write your code here
vector<int>v;int i=0,j=nums.size()-1,m=0;
/* if(nums.size()<2) return v;
multimap<int,int> m;int k=nums.size();
for(int i=0;i<k;++i)
m.insert({nums[i],i+1});
multimap<int,int>::iterator it=m.begin(),it2=m.end(),it3=it2,it4=it,it5=it2;
if(it->first>target) return v;
for(;it!=it2;++it){
k=it->first;
if(k<=target)
{
for(it4=++it;it4!=it5;++it4){
if(it4->first==target-k)
{int a=(--it)->second,b=it4->second;
// a<b?{a=a+b;b=a-b;a=a-b;}:{a=a;b=b;}
// if(a<b){a=a+b;b=a-b;a=a-b;}
// if(a<b){int temp=b;b=a;a=temp;}
v.push_back(a);
v.push_back(b);
return v;}
if(it4->first>target-k) {it5=it4;break;}
}
}
else break;
}*/
sort(nums.begin(),nums.end());
while(i<j){
m=nums[i]+nums[j];
if(m==target)
{ v.push_back(i+1);v.push_back(j+1);return v;}
if(m>target) j--;
if(m<target) i++;
}
return v;
}
};
可是错误,原因在于我把排序后的顺序当答案,题目要的是原顺序,而不是排序过后的顺序
之后,在我改正这个错误后,我又遇见一个错误,当nums[i]==nums[j]的时候,运用以下代码:
for(auto c:NUM){
p++;
if(c==nums[i]) v.push_back(p);
if(c==nums[j]) v.push_back(p);}
测试数据是0,1,2,0
答案应该是1,4
但是输出的确是1,1,4,4
改正后正确代码:
class Solution {
public:
/*
* @param numbers : An array of Integer
* @param target : target = numbers[index1] + numbers[index2]
* @return : [index1+1, index2+1] (index1 < index2)
*/
vector<int> twoSum(vector<int> &nums, int target) {
// write your code here
vector<int>v;int i=0,j=nums.size()-1,m=0;
/* if(nums.size()<2) return v;
multimap<int,int> m;int k=nums.size();
for(int i=0;i<k;++i)
m.insert({nums[i],i+1});
multimap<int,int>::iterator it=m.begin(),it2=m.end(),it3=it2,it4=it,it5=it2;
if(it->first>target) return v;
for(;it!=it2;++it){
k=it->first;
if(k<=target)
{
for(it4=++it;it4!=it5;++it4){
if(it4->first==target-k)
{int a=(--it)->second,b=it4->second;
// a<b?{a=a+b;b=a-b;a=a-b;}:{a=a;b=b;}
// if(a<b){a=a+b;b=a-b;a=a-b;}
// if(a<b){int temp=b;b=a;a=temp;}
v.push_back(a);
v.push_back(b);
return v;}
if(it4->first>target-k) {it5=it4;break;}
}
}
else break;
}*/
vector<int> NUM(nums);int p=0,q=0;
sort(nums.begin(),nums.end());
while(i<j){
m=nums[i]+nums[j];
if(m==target)
// { v.push_back(i+1);v.push_back(j+1);return v;}
break;
if(m>target) j--;
if(m<target) i++;
}
for(auto c:NUM){
p++;
if(c==nums[i]) v.push_back(p);
if(nums[i]!=nums[j]){
if(c==nums[j]) v.push_back(p);}}
return v;
}
};
之后,我看到更简洁的方案:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> m;
for (int i = 0; i < nums.size(); ++i) {
if (m.count(target - nums[i])) {
return {i, m[target - nums[i]]};
}
m[nums[i]] = i;
}
return {};
}
};