[BZOJ 4589]Hard Nim

Description

题库链接

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

\(1\leq n\leq 10^9,2\leq m\leq 50000\)

Solution

我们记多项式 \(A(x)\) ，对于 \(x_i\) 若 \(i\leq 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\n", a[0]);
    }
}
int main() {work(); return 0; }