BZOJ4347 [POI2016] Nim z utrudnieniem

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

首先复杂度上如果有n那么一定是会TLE的。

然后突破口就是m<=10000000

然后把a从小到大排序，每次只需要枚举0~2*a[i]进行转移。

毕竟sigma(a[i])<=10000000

再说原地转移的技巧，好。。。。。。

ACcode:

#include<cstdio>
#include<cstring>
#include<cctype>
#include<algorithm>
#define mod 1000000007
using namespace std;

const int N=(1<<20)+50;
int n,d,a[500005],dp[10][N+20],tmp[N+20],sum;

int main()
{
	scanf("%d%d",&n,&d);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),sum^=a[i];
	sort(a+1,a+1+n);
	dp[0][0]=1;
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<=min(N,2*a[i]);j++) tmp[j]=(dp[0][j]+dp[d-1][j^a[i]])%mod;
		for(int j=d-1;j>0;j--) 
			for(int k=0;k<=min(N,2*a[i]);k++)
				if(k<=(a[i]^k))
				{
					int tmp1=dp[j-1][k],tmp2=dp[j-1][a[i]^k];
					dp[j][k]=(dp[j][k]+tmp2)%mod;
					dp[j][k^a[i]]=(dp[j][k^a[i]]+tmp1)%mod;
				}
		for(int j=0;j<=min(N,2*a[i]);j++) dp[0][j]=tmp[j];
	}
	
	if(n%d==0) dp[0][sum]=(dp[0][sum]+mod-1)%mod;
	printf("%d\n",dp[0][sum]);
}