Sumsets UVA - 10125

 

一个集合中需要找到 a,b,c,d ,使得 a+b+c=d

 

枚举a,b ，计算a+b，存在map里

再枚举d,c ，计算d-c 是否 存在 d-c==a+b

 

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46

#include <iostream> #include <map> #include <algorithm> #include <vector> using namespace std; const int N=1e3+2;   //a+b==d-c #define pii pair<int,int> #define mk(x,y) make_pair(x,y) int n;  map<int,vector<pii> > mp;  int a[N];     int H(){     int i,j;     for(j=n;j>0;j--)      for(i=j-1;i>0;i--){         int t=a[j]-a[i];         for(int k=0;k<mp[t].size();k++){             int xx=mp[t][k].first,yy=mp[t][k].second;             if(xx!=a[i]&&yy!=a[j]&&xx!=a[j]&&yy!=a[i])                 return j;         }      }     return -1;  }  void sov(){     int i,j;     for(i=1;i<=n;++i) cin>>a[i];sort(a+1,a+1+n);     for(i=1;i<=n;i++)      for(j=i+1;j<=n;j++)       mp[a[i]+a[j]].push_back(mk(a[i],a[j]));           int t=H();     if(t==-1) cout<<"no solution"<<endl;     else  cout<<a[t]<<endl;  }  signed main(){     while(cin>>n,n){         mp.clear() ;         sov();     }  }