【POI2016】【BZOJ4347】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],…,an。
本题中m不直接给出，但是保证m<=10000000。
Output

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

5 2

1 3 4 1 2
Sample Output

2
HINT

Source

鸣谢Claris

又是Nim游戏的变形.如果没有一开始那个拿走石子堆就是普通的Nim游戏了.所以要按SG值DP.
POI2016的题果然厉害…想了想发现还是不会做..然后打开了Claris的博客TAT
传送门
没想到DP时候数组两维的顺序在这题对速度影响那么大..
把两维反过来就从27s->14s

//27s
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define GET (ch>='0'&&ch<='9')
#define MAXN 1100000
#define P 1000000007
#define Mod(x)  x<P?x:x-P
using namespace std;
int n,d,m,p;
int cnt[MAXN];
unsigned int f[MAXN][10],g[MAXN],tmp;
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
int main()
{
    in(n);in(d);f[0][0]=1;p=1;int x;
    for (int i=1;i<=n;i++)  in(x),cnt[x]++,m=max(m,x);
    for (int i=1;i<=m;i++)
    {
        while (p<=i)    p<<=1;
        while (cnt[i]--)
        {
            for (int j=0;j<p;j++)   g[j]=Mod((f[j][d-1]+f[i^j][0]));
            for (int j=d-1;j;j--)
                for (int k=0;k<p;k++)
                    if (k<=(k^i))
                    {
                        tmp=f[k][j];
                        f[k][j]=Mod((f[k][j-1]+f[k^i][j]));f[k^i][j]=Mod((f[k^i][j-1]+tmp));
                    }
            for (int j=0;j<p;j++)   f[j][0]=g[j];
        }
    }
    printf("%u\n",Mod((f[0][0]+(P-(n%d==0)))));
}

//14s
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define GET (ch>='0'&&ch<='9')
#define MAXN 1100000
#define P 1000000007
#define Mod(x)  x<P?x:x-P
using namespace std;
int n,d,m,p;
int cnt[MAXN];
unsigned int f[10][MAXN],g[MAXN],tmp;
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
int main()
{
    in(n);in(d);f[0][0]=1;p=1;int x;
    for (int i=1;i<=n;i++)  in(x),cnt[x]++,m=max(m,x);
    for (int i=1;i<=m;i++)
    {
        while (p<=i)    p<<=1;
        while (cnt[i]--)
        {
            for (int j=0;j<p;j++)   g[j]=Mod(f[d-1][j]+f[0][i^j]);
            for (int j=d-1;j;j--)
                for (int k=0;k<p;k++)
                    if (k<=(k^i))
                    {
                        tmp=f[j][k];
                        f[j][k]=Mod(f[j-1][k]+f[j][k^i]);f[j][k^i]=Mod(f[j-1][k^i]+tmp);
                    }
            for (int j=0;j<p;j++)   f[0][j]=g[j];
        }
    }
    printf("%u\n",Mod(f[0][0]+(P-(n%d==0))));
}