首页 > 代码库 > 3Sum Closest从数列中找到三个数之和最接近给定值

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

即求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从数列中找到三个数之和最接近给定值