3Sum Closest從數(shù)列中找到三個數(shù)之和最接近給定值
來源:程序員人生 發(fā)布時間:2014-12-07 10:30:33 閱讀次數(shù):2735次
即求min{ target - a -b -c } a,b,c blog to Set S;
(1)最簡單的做法固然是求出所有的不相同的3個數(shù)和,保存到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)有了簡單解法,我們自然要優(yōu)化。
(1)3重循環(huán)肯定是可以干掉的,即先把兩數(shù)之和和它們對應的下標存到map里。
(2)然后再用數(shù)組的數(shù)去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++;
}
}
生活不易,碼農(nóng)辛苦
如果您覺得本網(wǎng)站對您的學習有所幫助,可以手機掃描二維碼進行捐贈