[BZOJ 4589]Hard Nim

Description

题库链接

两人玩 (nim) 游戏，(n) 堆石子，每堆石子初始数量是不超过 (m) 的质数，那么后手必胜的方案有多少种。对 (10^9+7) 取模。

(1leq nleq 10^9,2leq mleq 50000)

Solution

我们记多项式 (A(x)) ，对于 (x_i) 若 (ileq m) 且 (i) 为质数，那么 (x_i) 的系数为 (1) ，其余情况系数为 (0) 。

显然当 (n=2) 时，令多项式

[C(x)=A(x)oplus A(x)]

其中 (oplus) 为按位异或。那么 (c_0) 即为答案。

更多地， (C(x)=A^n(x)) ，那么 (c_0) 就是 (n) 堆石子的方案数。

(FWT) 乱搞一下即可。

Code

#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9+7, N = 50000;

int isprime[N+5], prime[N+5], tot, n, m, len, inv2;
int f[N*2+5], a[N*2+5];

int quick_pow(int a, int b) {
    int ans = 1;
    while (b) {
    if (b&1) ans = 1ll*ans*a%mod;
    b >>= 1, a = 1ll*a*a%mod;
    }
    return ans;
}
void get_prime() {
    memset(isprime, 1, sizeof(isprime)); isprime[1] = 0;
    for (int i = 2; i <= N; i++) {
    if (isprime[i]) prime[++tot] = i;
    for (int j = 1; j <= tot && i*prime[j] <= N; j++) {
        isprime[i*prime[j]] = 0;
        if (i%prime[j] == 0) break;
    }
    }
}
void FWT(int *A, int o) {
    for (int i = 1; i < len; i <<= 1)
    for (int j = 0; j < len; j += (i<<1))
        for (int k = 0; k < i; k++) {
        int x = A[k+j], y = A[k+j+i];
        A[k+j] = (x+y)%mod, A[k+j+i] = (x-y+mod)%mod;
        if (o == -1) A[k+j] = 1ll*A[k+j]*inv2%mod, A[k+j+i] = 1ll*A[k+j+i]*inv2%mod;
        }
}
void work() {
    inv2 = quick_pow(2, mod-2);
    get_prime();
    while (~scanf("%d%d", &n, &m)) {
    memset(f, 0, sizeof(f));
    for (int i = 1; i <= tot && prime[i] <= m; i++) f[prime[i]] = 1;
    memset(a, 0, sizeof(a)); a[0] = 1;
    for (len = 1; len <= m; len <<= 1);
    FWT(a, 1), FWT(f, 1);
    while (n) {
        if (n&1) for (int i = 0; i < len; i++) a[i] = 1ll*a[i]*f[i]%mod;
        for (int i = 0; i < len; i++) f[i] = 1ll*f[i]*f[i]%mod;
        n >>= 1;
    }
    FWT(a, -1); printf("%d
", a[0]);
    }
}
int main() {work(); return 0; }