2019牛客暑期多校训练营（第四场）E triples II —— DP+容斥

This way

题意：

让你选出n个3的倍数的序列使得这些数的或是a，问你有多少种可能。

题解：

能够DP是因为有一个性质：
两个数模3=1的数与模3=2的数相同时，子集中一些数异或成为它的子集的方案数是一样的。
一个数的是另一个数的子集当且仅当这个数与上另一个数之后是它本身。
那么定义$dp[i][j]$表示这个数的2进制位上%3=1的位有i个，%3=2的位有j个时，它子集是3的倍数的数量，比如0011就有2个有效子集：0011和0000。那么dp[1][1]=2。
那么$dp[i][j]={\sum }_{i=0}^{N-1}{\sum }_{j=0}^{N-1}({\sum }_{x=0}^i{C}_i^x\ast {\sum }_{y=0}^j{C}_j^y)$
也就是枚举总共有多少位的同时枚举当前有多少位。
至于$(nu{m}_o+nu{m}_t-i-j)\%2\&\&(f\ast =-1);$就是容斥了
我们加上所有$nu{m}_o+nu{m}_t$的情况，减去$nu{m}_o+nu{m}_t-1$的情况，再加上…
为什么是容斥，因为减去dp是a的子集的时候，它们会有交集比如dp[2][2]-dp[2][1]-dp[1][2]，dp[2][1]和dp[1][2]会有交集，所以要在下面加回来

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll mod=998244353;
const int N=64;
ll c[N][N],dp[N][N];
ll qpow(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1)
            ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void init()
{
    for(int i=0;i<N;i++)
        c[i][0]=1;
    for(int i=1;i<N;i++)
        for(int j=1;j<=i;j++)
            c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod;
    //如果两个数在s中的第一维和第二维相同，那么他们的方案数就相同：1001=0011
    for(int i=0;i<N;i++)
        for(int j=0;j<N;j++)
            for(int x=0;x<=i;x++)
                for(int y=0;y<=j;y++)
                    (x+2*y)%3||(dp[i][j]=(dp[i][j]+c[j][y]*c[i][x])%mod);
    dp[0][0]=1;
}
int main()
{
    init();
    int t;
    scanf("%d",&t);
    while(t--)
    {
        ll n,a;
        scanf("%lld%lld",&n,&a);
        ll ans=0;
        int num_o=0,num_t=0;
        for(int i=0;i<N;i++)
            if(((a>>i)&1))
                i%2&&(num_o++),i%2||(num_t++);
        for(int i=0;i<=num_o;i++)
        {
            for(int j=0;j<=num_t;j++)
             {
                 ll f=c[num_o][i]*c[num_t][j]%mod*qpow(dp[i][j],n)%mod;
                 (num_o+num_t-i-j)%2&&(f*=-1);
                 ans=(ans+f+mod)%mod;
             }
        }
        printf("%lld\n",(ans+mod)%mod);
    }
    return 0;
}