从今天开始刷leetcode,记录我做题的每一步,有错误希望大家指出,谢谢
3Sum
最简单,但是没效率,没有通过
class Solution {
public:
vector<vector<int> > threeSum(vector<int> &num) {
int i,j,k;
vector<vector<int>> sum;
sum.clear();
if(num.size()<3)
return sum;
sort(num.begin(),num.end());
for(i=0;i<num.size()-2;i++)
{
if(num[i]>0)
break;
for(j=i+1;j<num.size()-1;j++)
{
if(num[i]+num[j]>0)
break;
for(k=j+1;k<num.size();k++)
{
if(num[i]+num[j]+num[k]>0)
break;
if((num[i]+num[j]+num[k])==0)
{
vector<int> temp;
temp.clear();
temp.push_back(num[i]);
temp.push_back(num[j]);
temp.push_back(num[k]);
sum.push_back(temp);
}
}
}
}
return sum;
}
};
报了个错 Submission Result: Output Limit Exceeded
百度了一下这个错误,坑爹,说是死循环,看了半天的代码,没有问题呀,我就纳闷了,搞了半天没弄出来,最后用工具编译了一下,才发现没有去重,leetcode直接报错,无语,加了一句代码
sum.erase(unique(sum.begin(), sum.end()), sum.end() );
Last executed input: | [5,-9,-11,9,9,-4,14,10,-11,1,-13,11,10,14,-3,-3,-4,6,-15,6,6,-13,7,-11,-15,10,-8,13,-14,-12,12,6,-6,8,0,10 |
总算是没错了,不过三层循环效率太低,只能换种高效的算法
class Solution {
public:
vector<vector<int> > threeSum(vector<int> &num) {
vector<vector<int>> returnNum;
sort(num.begin(),num.end());
for(int i=0;i!=num.size();i++)
{
if(num[i]>0)
break;
if(i>0&&num[i]==num[i-1])
continue;
int j,k;
j=i+1;
k=num.size()-1;
while(j<k)
{
if(j>i+1&&num[j]==num[j-1])
{
j++;
continue;
}
if(k<num.size()-1&&num[k]==num[k+1])
{
k--;
continue;
}
int sum = num[i]+num[j]+num[k];
if(sum>0)
k--;
else if(sum<0)
j++;
else
{
vector<int> tmp;
tmp.push_back(num[i]);
tmp.push_back(num[j]);
tmp.push_back(num[k]);
returnNum.push_back(tmp);
j++;
}
}
}
return returnNum;
}
};
成功