bzoj4589

fwt

原理并不知道

nim游戏石子异或和=0后手赢

那么也就是求a[1]^a[2]^...^a[n]=0的方案数

这个和bzoj3992一样可以dp

dp[i][j]表示前i个数异或和为j的方案数 dp[0][0] = 1

dp[i][j] = dp[i - 1][k] * a[p] p ^ k = j a[p] = 0 / 1 表示有没有p这个数

这个东西也不能矩阵快速幂

但是我们有一个叫fwt的东西

能够求c = a @ b @是一种运算 -----> c[i] = a[j] * b[k] i = j ^ k

那么每次转移就是fwt了

由于转移的形式一样 那么就可以快速幂 并且由于fwt运算并不会向fft那样下标多出来一些东西 也就不用算循环卷积

直接fwt后每个数pow一下 再ifwt就行了

复杂度O(n log n + n log m)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = ;
const ll P = ;
int n, m, len;
ll inv;
ll a[N];
int p[N], mark[N];
ll power(ll x, ll t)
{
    ll ret = ;
    for(; t; t >>= , x = x * x % P) if(t & ) ret = ret * x % P;
    return ret;
}
void fwt(ll *a, int n)
{
    for(int l = ; l <= n; l <<= )
    {
        int m = l >> ;
        for(int i = ; i < n; i += l)
            for(int k = ; k < (l >> ); ++k)
            {
                ll x = a[i + k], y = a[i + k + m];
                a[i + k] = (x + y) % P;
                a[i + k + m] = ((x - y) % P + P) % P;
            }
    }
}
void ifwt(ll *a, int n)
{
    for(int l = n; l >= ; l >>= )
    {
        int m = l >> ;
        for(int i = ; i < n; i += l)
            for(int k = ; k < m; ++k)
            {
                ll x = a[i + k], y = a[i + m + k];
                a[i + k] = (x + y) % P * inv % P;
                a[i + m + k] = ((x - y) % P + P) % P * inv % P;
            }
    }
}
int main()
{
    inv = power(, P - );
    for(int i = ; i <= ; ++i)
    {
        if(!mark[i]) p[++p[]] = i;
        for(int j = ; j <= p[] && i * p[j] <= ; ++j)
        {
            mark[i * p[j]] = ;
            if(i % p[j] == ) break;
        }
    }
    while(scanf("%d%d", &n, &m) != EOF)
    {
        memset(a, , sizeof(a));
        for(int i = ; i <= p[] && p[i] <= m; ++i) a[p[i]] = ;
        for(len = ; len <= m; len <<= );
        fwt(a, len);
        for(int i = ; i < len; ++i) a[i] = power(a[i], n);
        ifwt(a, len);
        printf("%lld\n", a[]);
    }
    return ;
}