Gold Balanced Lineup

  1 #include<cstdio>
  2 #include<cstring>
  3 #include<cmath>
  4 #include <cstdlib>
  5 #define MAXN 100001
  6 using namespace std;
  7 
  8 const int mod=99991;
  9 int sum[MAXN][30];
 10 int f[MAXN][30];
 11 int c[MAXN][30];
 12 int n,k,len;
 13 struct node
 14 {
 15     int xx;
 16     node *next;
 17 };
 18 node *hash[mod];
 19 
 20 bool cmp(int a,int b)
 21 {
 22     for(int j=0; j<k; j++)
 23         if(c[a][j]!=c[b][j])
 24             return false;
 25     return true;
 26 }
 27 
 28 void insert(int x)
 29 {
 30     int key=0;
 31     for(int j=1; j<k; j++)
 32     {
 33         key+=c[x][j]*j;
 34     }
 35     key=abs(key)%mod;
 36     if(!hash[key])
 37     {
 38         node *p1=new node();
 39         p1->xx=x;
 40         hash[key]=p1;
 41     }
 42     else
 43     {
 44         node *p=hash[key];
 45         if(cmp(p->xx,x))
 46         {
 47             int d=x-(p->xx);
 48             if(len<d)
 49                 len=d;
 50             return;
 51         }
 52         else
 53         {
 54             while(p->next)
 55             {
 56                 if(cmp(p->next->xx,x))
 57                 {
 58                     int dd=x-(p->next->xx);
 59                     if(dd>len)
 60                         len=dd;
 61                     return;
 62                 }
 63                 p=p->next;
 64             }
 65             node *pp=new node();
 66             pp->xx=x;
 67             p->next=pp;
 68         }
 69     }
 70     return ;
 71 }
 72 
 73 void inti()
 74 {
 75     for(int i=0; i<k; i++)
 76     {
 77         c[0][i]=0;
 78         sum[0][i]=0;
 79     }
 80     for(int i=0; i<mod; i++)
 81         hash[i]=NULL;
 82 }
 83 
 84 int main()
 85 {
 86     while(scanf("%d%d",&n,&k)!=EOF)
 87     {
 88         inti();
 89         insert(0);
 90         len=0;
 91         int m;
 92         for(int i=1; i<=n; i++)
 93         {
 94             scanf("%d",&m);
 95             for(int j=0; j<k; j++)
 96             {
 97                 f[i][j]=m%2;
 98                 m/=2;
 99                 sum[i][j]=sum[i-1][j]+f[i][j];
100                 c[i][j]=sum[i][j]-sum[i][0];
101             }
102             insert(i);
103         }
104         printf("%d\n",len);
105     }
106     return 0;
107 }

View Code