poj3274(数的哈希)

  特别好的一个题目，题目的大概意思是，每头牛都有一定的属性值，然后把这个属性值转化为一个二进制数，从右到左分别呢标号为1,2...N，如果某一位上为1，那么它就具有这个属性。现在定义一个区间是“balanced”，当它满足区间内每个属性出现的次数相同，要求这样的区间的最大长度。

  

  分析：令sum[i][j]代表前i头牛j属性的个数和，那么balanced的区间（i,j）就满足

  sum[j][0] - sum[i][0] = sum[j][1] - sum[i][1] =   .......  = sum[j][K-1] - sum[i][K-1]

  进一步可以转化为：

  sum[i][1] - sum[i][0] = sum[j][1] - sum[j][0]，

  sum[i][2] - sum[i][0] = sum[j][2] - sum[j][0]，

  ......

  ......

  sum[i][K-1] - sum[i][0] = sum[j][K-1] - sum[j][0]

  那么题目的意思记录每一个sum[i][j] - sum[i][0], j = 1......K-1的值，很显然这个值相同的行，它们这个区间就是balanced区间。那么这种情况就是要用数的Hash了。

  然后需要注意的是，因为是减法，那么可能出现负值，所以给它加个值就好了，还有一个要注意的是，如果刚好sum[i]的中每一项都相等，那么前i项也是balanced区间。

  然后自己很傻的初始化ans为1，还有算出来的结果忘记了取Mod，RE了好几次，都是要注意的。

  

#include "stdio.h"
#include "string.h"
#include "math.h"
#include <string>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <algorithm>
#include <iostream>
using namespace std;

#define MAXM 1
#define MAXN 1
#define max(a,b) a > b ? a : b
#define min(a,b) a < b ? a : b
int Mod = 1000000007;
#define Mem(a,b) memset(a,b,sizeof(a))
double pi = acos(-1.0);
double eps = 1e-6;

int prime = 99991; //模的数目
int Inc = 100001;

struct Node{
	int num[32];
	int flag;
	struct Node *next;
};

Node node[100000];

int sum[2][32];
int K,N;

bool Comp(Node* b, int *p){
	int vv = p[0] - b->num[0];
	for(int i = 1; i < K; i ++){
		if( p[i] - b->num[i] != vv )
			return false;
	}
	return true;
}

void solve(){
	for(int i = 0; i < prime; i ++){
		node[i].flag = -1;
		node[i].next = NULL;
	}

	int val;
	Mem(sum,0);
	int ans = 0, ssum = 0;

	for(int i = 0; i < N; i ++){
		scanf("%d",&val);
		ssum = 0;
		int j = 0;
		if( i == 0 ){
			while( val ){
				if( val & 1 )
					sum[i][j] ++;
				val >>= 1;
				j ++;
			}

			for(j = 1; j < K; j ++){
				if( sum[i%2][j] != sum[i%2][0] )
					break;
			}
			if( j == K && ans < i + 1 )
				ans = i + 1;

			for(j = K - 1; j > 0; j --){
				ssum += (sum[i][j] - sum[i][0] + Inc) % prime;
			}
			ssum %= prime;
			if( node[ssum].flag == -1 ){
				node[ssum].flag = 0;
				node[ssum].next = new struct Node;
				for(j = 0; j < K; j ++){
					node[ssum].next->num[j] = sum[i][j];
				}
				node[ssum].next->flag = i;
				node[ssum].next->next = NULL;
			}
		}
		else{

			for(j = 0; j < K; j ++)
				sum[i%2][j] = sum[(i-1)%2][j];
			j = 0;
			while( val ){
				if( val & 1 )
					sum[i%2][j] = sum[(i-1)%2][j] + 1 ;
				val >>= 1;
				j ++;
			}

			for(j = 1; j < K; j ++){
				if( sum[i%2][j] != sum[i%2][0] )
					break;
			}
			if( j == K && ans < i + 1 )
				ans = i + 1;

			for(j = K - 1; j > 0; j --){
				ssum += (sum[i%2][j] - sum[i%2][0] + Inc) % prime;
			}
			ssum %= prime;
			if( node[ssum].flag == -1 ){
				node[ssum].flag = 0;
				node[ssum].next = new struct Node;
				for(j = 0; j < K; j ++){
					node[ssum].next->num[j] = sum[i%2][j];
				}
				node[ssum].next->flag = i;
				node[ssum].next->next = NULL;
			}
			else{
				Node *b = node[ssum].next, *a;
				while(b){
					a = b;
					if( Comp(b, sum[i%2]) ){
						if( ans < i - b->flag ){
							ans = i - b->flag ;
							break;
						}
					}
					b = b->next;
				}
				if( b == NULL ){
					a->next = new struct Node;
					for(j = 0; j < K; j ++){
						a->next->num[j] = sum[i%2][j];
					}
					a->next->flag = i;
					a->next->next = NULL;
				}
			}
		}
	}

	printf("%d\n",ans);
}


int main()
{
//	freopen("d:\\test.txt", "r", stdin);
	while(cin>>N){
		cin>>K;
		solve();
	}

	return 0;
}