POJ 3274 Gold Balanced Lineup

hash存储，查询，跟POJ 3349类似

#include<stdio.h> #include<stdlib.h>; #include<string.h> #define inf 100001 struct Edge{ int pos[32],next; }edge[100100]; int head[100100],max,num[100100],k; int Max(int a,int b){ return a>b?a:b; } int add(int b,int pos){ int i,j; int hash=b%inf; for(i=head[hash];i!=-1;i=edge[i].next){ if(b==num[i]){ bool flag=0; int len=edge[pos].pos[1]-edge[i].pos[1]; for(j=2;j<=k;j++) if(edge[pos].pos[j]-edge[i].pos[j]!=len){ flag=1; break; } if(!flag) return i; } } edge[pos].next=head[hash]; head[hash]=pos; return pos; } int main(){ int i,j,n,tem; memset(head,-1,sizeof(head)); scanf("%d %d",&n,&k); int mod=(1<<k)-1; for(i=1;i<=k;i++) edge[0].pos[i]=0; num[0]=0; max=0-add(0,0); for(i=1;i<=n;i++){ scanf("%d",&tem); num[i]=(num[i-1]+tem)%mod; for(j=1;j<=k;j++){ if(tem&(1<<(j-1))) edge[i].pos[j]=edge[i-1].pos[j]+1; else edge[i].pos[j]=edge[i-1].pos[j]; } int temp=add(num[i],i); max=Max(max,i-temp); } printf("%d\n",max); return 0; }