给定多头牛的属性,每头牛的属性由一个非负数表示,该数的二进制表示不会超过K位,它的二进制表示的每一位若为1则表示该牛有对应的第i种属性,若为0则表示没有该属性。
对于给定的牛的顺序,要求输出某一段子序列的长度,这个子序列中的牛的K个属性对应相加以后全部相等。
假设n=3, k = 3
输入的3个数变成的二进制分别为(a1, a2, a3), (b1, b2, b3), (c1, c2, c3)
设sum(i)为从第1个数到第i个数的属性和的序列
若从第2个数到第3个数的序列满足条件,则说明b1+c1 = b2+c2 = b3+c3,即sum(3)-sum(2)的序列每一位都相等
推广一下,若sum(i) = (a, b, c),sum(j) = (d, e, f),且i到j这个子序列满足条件,则说明(d, e, f) - (a, b, c) = (x, x, x),即(d, e, f) = (a + x, b + x, c + x)。每个序列中的数都减去序列中的最后一个数,得到(d - f, e - f, 0) = (a - c, b - c, 0)。因此只要判断两个完全转换过后的序列是否相同,就可以知道它们之间的原序列是否满足条件了。
所以解题的第一步是把原来的数转换为二进制序列,第二步是把二进制序列转换成sum序列,即逐步叠加,第三步是把每个sum序列都减去该序列的最后一个数,最后一步是把这些序列进行哈希,计算它们的最大差距。
有一点要注意,如果从第1个数到第i个数这段序列满足条件,即sum(i) - sum(0) = (x, x, x),则说明sum(i)的各个位都是相同的,因此需要在第三步之前先做这个判断,把符合条件的序列找出来,更新一下答案。
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 100010;
const int K = 32;
const int H = 100007;
int n, k, ans;
int num[N][K];
struct Node
{
int pos;
int next;
};
Node node[N];
int cur;
int hashTable[H];
bool cmp(int x1, int x2)
{
for (int i = 0; i < k; ++i)
{
if (num[x1][i] != num[x2][i]) return false;
}
return true;
}
bool allsame(int x)
{
for (int i = 0; i < k - 1; ++i)
{
if (num[x][i] != num[x][i + 1]) return false;
}
return true;
}
void init()
{
for (int i = 0; i < H; ++i) hashTable[i] = -1;
ans = 0;
cur = 0;
}
unsigned int getHash(int x)
{
unsigned int hash = 0;
for (int i = 0; i < k; ++i)
{
hash += num[x][i] * (i + 1);
}
return (hash & 0x7fffffff) % H;
}
void searchHash(int x)
{
unsigned int h = getHash(x);
int next = hashTable[h];
while (next != -1)
{
if (cmp(node[next].pos, x))
{
if (x - node[next].pos > ans) ans = x - node[next].pos;
return;
}
next = node[next].next;
}
node[cur].pos = x;
node[cur].next = hashTable[h];
hashTable[h] = cur;
++cur;
}
int main()
{
int val;
init();
scanf("%d%d", &n, &k);
for (int i = 0; i < n; ++i)
{
scanf("%d", &val);
for (int j = 0; j < k; ++j)
{
num[i][j] = (val & 1);
val >>= 1;
}
}
for (int i = 1; i < n; ++i)
{
for (int j = 0; j < k; ++j)
{
num[i][j] += num[i - 1][j];
}
}
for (int i = n - 1; i >= 0; --i)
{
if (allsame(i))
{
ans = i + 1;
break;
}
}
for (int i = 0; i < n; ++i)
{
for (int j = 0; j < k; ++j)
{
num[i][j] -= num[i][k - 1];
}
}
for (int i = 0; i < n; ++i) searchHash(i);
printf("%d\n", ans);
return 0;
}