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

来源:互联网 发布:php开发游戏服务器 编辑:程序博客网 时间:2024/04/29 16:59

即求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++;     }}


0 0