3Sum Closest从数列中找到三个数之和最接近给定值

标签：style   blog   ar   os   sp   for   strong   on   log   

即求min{ target - a -b -c } a,b,c blog to Set S;

(一)最简单的做法当然是求出所有的不相同的三个数和，保存到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++;
     }
}

(二)有了简单解法，我们自然要优化。

（1）三重循环肯定是可以干掉的，即先把两数之和和它们对应的下标存到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++;
     }
}

3Sum Closest从数列中找到三个数之和最接近给定值

标签：style   blog   ar   os   sp   for   strong   on   log   

原文地址：http://blog.csdn.net/ylzintsinghua/article/details/41730449