POJ 3274 Gold Balanced Lineup (前缀和 + 哈希表)

大致题意:  给出n只牛和m个特征点,  给每只牛一个编号，这个编号的m位二进制数表示具有某些特征, 例如 3 => 011 表示该牛具有第1和第2个特征, 现在要求,最长的一段区间内使得将每只牛拥有的各种特征累加起来得到的每个特征的总数都相等,求区间长度

中文题目:

以下分析由  博客:優YoU  给出

分析:

数组sum[i][j]表示从第1到第i头cow属性j的出现次数。

所以题目要求等价为:

求满足

sum[i][0]-sum[j][0]=sum[i][1]-sum[j][1]=.....=sum[i][k-1]-sum[j][k-1] (j<i)

中最大的i-j

 

将上式变换可得到

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]

 

令C[i][y]=sum[i][y]-sum[i][0] (0<y<k)

初始条件C[0][0~k-1]=0

 

所以只需求满足C[i][]==C[j][] 中最大的i-j，其中0<=j<i<=n。

C[i][]==C[j][] 即二维数组C[][]第i行与第j行对应列的值相等，

那么原题就转化为求C数组中 相等且相隔最远的两行的距离i-j。

 

以样例为例

7 3

7

6

7

2

1

4

2

先把7个十进制特征数转换为二进制，并逆序存放到特征数组feature[ ][ ]，得到：

7 à 1 1 1

6 à 0 1 1

7 à 1 1 1

2 à 0 1 0

1 à 1 0 0

4 à 0 0 1

2 à 0 1 0

（行数为cow编号，自上而下从1开始；列数为特征编号，自左到右从0开始）

 

再求sum数组，逐行累加得，sum数组为

 1 1 1

1 2 2

2 3 3

2 4 3

3 4 3

3 4 4

3 5 4

再利用C[i][y]=sum[i][y]-sum[i][0]求C数组，即所有列都减去第一列

注意C数组有第0行，为全0

 0 0 0 à 第0行

0 0 0

0 1 1

0 1 1

0 2 1

0 1 0

0 1 1

0 2 1

显然第2行与第6行相等，均为011，且距离最远，距离为6-2=4，这就是所求。

/*******************************************************************************************/

Code:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn = 1e5+50;
int n, k;
int sum[maxn][40];
int HASH[maxn];
/**
 * [getHASH description]
 * @return     hash-key
 */
int getHASH(int num[]){
	int key = 0;
	for(int i=1; i<=k; i++){
		key = (key*7 + ((num[i])<<2))%maxn;
	}
	return abs(key)%maxn;
}
int main()
{
	//freopen("in.txt","r", stdin);
	int i, j, x, p, MAX=0;
	scanf("%d %d", &n, &k);
	memset(HASH, -1, sizeof(HASH));
	HASH[0]=0;//第0只牛默认为全0， 先放入哈希表中
	for(i=1; i<=n; i++){
		scanf("%d", &x);
		/**
		 * 1. 十进制转化称二进制, 反向储存
		 * 2. 每一列都减去第一列
		 *    WHY? reason: 减去增量才能进行判断
		 */
		for(j=1; j<=k; j++){
			sum[i][j] = sum[i-1][j] + (x&1);
			x>>=1;
		}
		p = sum[i][1];
		for(j=1; j<=k; j++){
			sum[i][j]-=p;
		}
		// for(j=1; j<=k; j++){
		// 	printf("%d ", sum[i][j] );
		// }
		//printf("\n");
		/**
		 * 利用HASH表去搜索在i之前最远地方与i列值相同的位置
		 * 注意,处理冲突情况
		 */
		int key = getHASH(sum[i]);
		//cout <<"key"<< key <<" " << HASH[key]<<endl;
		while(HASH[key]!=-1){
			for(p=1; p<=k; p++){
				if(sum[i][p]!=sum[HASH[key]][p]) break;
			}
			if(p>k){
				MAX = max(MAX, i-HASH[key]);
				break;
			}
			key++;
			key%=maxn;
		}
		if(HASH[key]==-1){
			HASH[key]=i;
		}
	}
	printf("%d\n", MAX);
	return 0;
}