4347: [POI2016]Nim z utrudnieniem

4347: [POI2016]Nim z utrudnieniem

Time Limit: 30 Sec   Memory Limit: 64 MB
Submit: 274   Solved: 124
[ Submit][ Status][ Discuss]

Description

A和B两个人玩游戏，一共有m颗石子，A把它们分成了n堆，每堆石子数分别为a[1],a[2],...,a[n]，每轮可以选择一堆石子，取掉任意颗石子，但不能不取。谁先不能操作，谁就输了。在游戏开始前，B可以扔掉若干堆石子，但是必须保证扔掉的堆数是d的倍数，且不能扔掉所有石子。A先手，请问B有多少种扔的方式，使得B能够获胜。

Input

第一行包含两个正整数n,d(1<=n<=500000,1<=d<=10)。

第二行包含n个正整数a[1],a[2],...,a[n](1<=a[i]<=1000000)。

本题中m不直接给出，但是保证m<=10000000。

Output

输出一行一个整数，即方案数对10^9+7取模的结果。

Sample Input

5 2
1 3 4 1 2

Sample Output

2

HINT

Source

鸣谢Claris

[ Submit][ Status][ Discuss]



Nim游戏的结论，，，如果n堆石子xor和为0后手胜否则先手胜

那么问题就转换成，有多少种方案，取走k堆石子，d|k && k != n，剩余石子xor和为0。

后面的就想不到了。。。。。。

先将石子按照每堆的大小，从小到大排序

一个数字x，假设它有k位，那么x与任意小于等于x的数字随意xor，结果严格小于1<<k

因此，定义状态f[i][j][k]:前i堆石子，取走的堆数mod d为j，没取走的xor和为k的方案数

显然，f[i][j][k] = f[i-1][j-1][k] + f[i-1][j][k^a[i]]，a[i]为第i堆石子的数量

由于m = ∑a[i] <= 1E7，第i堆石子转移的复杂度为O(2*d*a[i])，所以总复杂度近似O(m*d)

但是上面是个轻易MLE的方程。。。显然的优化是滚动数组

不过内存只给64MB，滚动数组也要80MB的内存。。。。

但，，，f[i][j][k^a[i]] = f[i-1][j-1][k^a[i]] + f[i-1][j][k]

也就是说这两个方程互相利用对方的先前状态，可以一起转移，就不用滚动数组了。。

不过f[i][0][op]这样的就不行了，，这里单独转移

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
using namespace std;

const int mo = 1000000007;
const int maxn = (1<<20);

int n,m,d,c[maxn],f[10][maxn],g[maxn];

int Add(const int &x,const int &y) {return (x + y) % mo;}

int getint()
{
	char ch = getchar(); int ret = 0;
	while (ch < '0' || '9' < ch) ch = getchar();
	while ('0' <= ch && ch <= '9')
		ret = ret*10 + ch - '0',ch = getchar();
	return ret;
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	n = getint(); d = getint(); f[0][0] = 1;
	for (int i = 1; i <= n; i++)
	{
		int w = getint();
		++c[w]; m = max(m,w);
	}
	for (int k = 1; k <= m; k++)
		while (c[k]--)
		{
			int p = 1; while (p <= k) p <<= 1;
			for (int i = 0; i < p; i++)
				g[i] = Add(f[d-1][i],f[0][i^k]);
			for (int i = d - 1; i; i--)
				for (int j = 0; j < p; j++)
				{
					if (j > (j^k)) continue;
					int t = f[i][j];
					f[i][j] = Add(f[i-1][j],f[i][j^k]);
					f[i][j^k] = Add(f[i-1][j^k],t);
				}
			for (int i = 0; i < p; i++) f[0][i] = g[i];
		}
	if (n % d == 0) f[0][0] = Add(f[0][0],mo - 1);
	cout << f[0][0];
	return 0;
}