poj 2549 Sumsets

hash应用。

a+b+c=d,各个值都是不等的，用四个循环肯定超时，所以把c移到右边，则变成两个两重循环。把数存于s[]数组中。因为这些数比较大，所以把取模作为hash函数,关键值就是abs(s[i]+s[j])%N,并把i与j存于res数组中，hash表中存放的是res数组的下标。判断成立的条件：h=res[ptr->index].i
      k=res[ptr->index].j
      s[h]+s[k]==s[i]-s[j]&&i!=h&&i!=k&&j!=h&&j!=k

#include<iostream> #include<cmath> #include<cstdlib> #define N 19997 using namespace std; struct Hash { int index; Hash *next; }; Hash *ys[N]; struct result { int i,j; }res[500005]; int s[1005]; int cmp(const void *a,const void *b) { return *((int *)a)-*((int *)b); } int main() { int n,i,j,temp,m,h,k; Hash *ptr; bool find; while(scanf("%d",&n)) { if(n==0) break; find=m=0; for(i=1;i<=n;i++) scanf("%d",s+i); qsort(s+1,n,sizeof(s[1]),cmp); for(i=0;i<N;i++) ys[i]=NULL; for(i=1;i<n;i++) for(j=i+1;j<=n;j++) { temp=abs(s[i]+s[j])%N; res[m].i=i; res[m].j=j; ptr=new Hash; ptr->index=m++; ptr->next=ys[temp]; ys[temp]=ptr; } for(i=n;i>1;i--) { if(find) break; for(j=i-1;j>=1;j--) { if(find) break; temp=abs(s[i]-s[j])%N; ptr=ys[temp]; while(ptr) { h=res[ptr->index].i; k=res[ptr->index].j; if(s[h]+s[k]==s[i]-s[j]&&i!=h&&i!=k&&j!=h&&j!=k) { find=true; break; } ptr=ptr->next; } } } if(find) printf("%d/n",s[i+1]); else printf("no solution/n"); } return 0; }