POJ 3274 Gold Balanced Lineup 数组Hash

点击打开链接

题意：

思路：sum[i][j]前i头牛的feature j的总数。

sum[a][j]-sum[b][j]=sum[a][j-1]-sum[b][j-1].....=sum[a][0]-sum[b][0]
求出a b的差即可。
上式再转化：sum[a][j]-sum[a][j-1]=sum[b][j]-sum[b][j-1]......
设c[i][j]=sum[i][j]-sum[i][0];
c[a][j]==c[b][j]

然后对c[i]数组哈希，求出最大的b-a即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#define INF 0x3f3f3f3f
#define bug puts("*****************")
#define Mod 199997
using namespace std;
const int NN=100010;
int cow[NN][35];
int sum[NN][35];
int arr[NN][35];
int n,k,t;
//int Build(int a[]){
//    int sum=0;
//    for(int i=1;i<=k;i++){
//        sum=((sum<<1)+a[i]+Mod)%Mod;
//        //cout<<sum<<endl;
//    }
//    return sum;
//}
int judge(int x1,int x2){
    for(int i=1;i<=k;i++){
        if(arr[x1][i]!=arr[x2][i])
            return 0;
    }
    return 1;
}
vector<int>Hash[Mod+10];
int main(){
    memset(sum,0,sizeof(sum));
    memset(arr,0,sizeof(arr));

    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&t);
        for(int j=k;j>=1;j--){
            sum[i][j]=sum[i-1][j]+t%2;
            t/=2;
        }
    }

    for(int i=0;i<=n;i++){
    ///把i=0也加入Hash为了求当i为[1,n]中值时键值为0的情况出现可以统计此时的长度为i-0;
    ///否则会漏掉（1，i）为最大满足条件此时的键值为0的情况
        int ans=0;
        for(int j=1;j<=k;j++){
            arr[i][j]=sum[i][j]-sum[i][1];
            ans+=arr[i][j];
        }
         Hash[((ans+Mod)%Mod)].push_back(i);
    }

    int Max=0;
    for(int i=0;i<Mod;i++){
        for(int j=0;j<Hash[i].size();j++){
            for(int k=j+1;k<Hash[i].size();k++){
                if(judge(Hash[i][j],Hash[i][k])){
                    int temp=Hash[i][j]-Hash[i][k];
                    temp=(temp>0?(temp):(-1*temp));
                    if(temp>Max) Max=temp;
                }
            }
        }
    }
    printf("%d\n",Max);
    return 0;
}