3Sum Closest从数列中找到三个数之和最接近给定值
来源:程序员人生 发布时间:2014-12-07 10:30:33 阅读次数:2594次
即求min{ target - a -b -c } a,b,c blog to Set S;
(1)最简单的做法固然是求出所有的不相同的3个数和,保存到set里,然后用target,target (+/-) i ,i [0....]
复杂度基本上可以算是O(n^3)。
int threeSumClosest(vector<int> &num, int target) {
int n=num.size();
if(n<3) return 0;
unordered_set<int> sum;
for(int i=0 ; i<n; ++i)
for(int j=i+1;j<n; ++j)
for(int k=j+1; k<n;++k){
int key = num[i]+num[j]+num[k];
sum.insert(key);
}
int lap=0;
while(true){
if(sum.find(target+lap)!=sum.end()) return target+lap;
else if (sum.find(target-lap)!=sum.end()) return target-lap;
lap++;
}
}
(2)有了简单解法,我们自然要优化。
(1)3重循环肯定是可以干掉的,即先把两数之和和它们对应的下标存到map里。
(2)然后再用数组的数去map里映照,最初是a[i] +0 , 然后是a[i] +1 ,顺次增加,总会找到的。
(3)这样基本上算是O(N^2)了。
int threeSumClosest(vector<int> &num, int target) {
int n=num.size();
if(n<3) return 0;
unordered_map<int,pair<int,int>> num0;
for(int i=0 ; i<n; ++i)
for(int j=i+1;j<n; ++j){
int key = target-num[i]-num[j];
num0.insert(make_pair(key,make_pair(i,j)));
}
int lap=0;
while(true){
for(int i=0 ; i< num.size();++i){
if( num0.find(num[i]+lap)!=num0.end()
&& i!=num0[num[i]+lap].first
&&i!=num0[num[i]+lap].second
)
return target-lap;
else if(num0.find(num[i]-lap)!=num0.end()
&& i!=num0[num[i]-lap].first
&&i!=num0[num[i]-lap].second
)
return target+lap;
}
lap++;
}
}
生活不易,码农辛苦
如果您觉得本网站对您的学习有所帮助,可以手机扫描二维码进行捐赠