2019牛客暑期多校训练营（第四场）E triples II 容斥 求n个3的倍数或为a

题目链接： https://ac.nowcoder.com/acm/contest/884/E

题意：

你要构造一个 $1e18$ 的数组使得，每个数都是3的倍数并且所有的数或起来的结果为$a$ ，问你有多少个这样的数组。

做法：

我们将a的每一位二进制进行拆分，你会发现这些数 $1,2,4,8$ 在$mod3$之后分别是$1,2,1,2$，后面的数自然也遵从这样的规则。那么我们就可以将这个a的所有二进制为1和为2的位数分开考虑。

同时这里有一个结论（好吧就是我数学太菜感觉不到 ），任意个a的子集按位或出来的结果还是a的子集。由于直接对答案进行求解存在一定的难度，我们就可以把这个转化为对a的子集求子集，这个还是简单的，$S[i][j]$ 表示二进制位在模3之后有 $i$个1和$j$个2的时候的合法方案数。

$S[i][j]={\sum }_{x=0}^i{\sum }_{y=0}^j{C}_i^x{C}_j^y$ ，这个公式应该还是好理解的，即在$i$个1和$j$个2的时候，取这$i$个里面$x$个1,这$j$个里面$y$个2的方案数，求了个整个的子集。（即子集的子集！！！！）

然后我们就可以求答案了，答案不是简单的$S^n[nu{m}_1][nu{m}_2]$ ，而是要容斥做，为什么呢，应该可以想象的到，这个$S$里面还包括了很多可能使得其异或起来不能到$a$，而真正全部包括了的答案被包括在这个$S$里。所以我们要用容斥把它一点点筛出来。即最后的答案$ans={\sum }_{i=0}^{nu{m}_1}{\sum }_{j=0}^{nu{m}_2}(-1{)}^{nu{m}_1+nu{m}_2-i-j}{C}_{nu{m}_1}^i{C}_{nu{m}_2}^j$

代码

#include <bits/stdc++.h>
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int mod=998244353;
ll C[67][67],dp[67][67],n,a;
int n1,n2;


ll quick(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;
        b/=2;
    }
    return ans;
}
void init(){
    rep(i,0,64) C[i][0]=1;
    rep(i,1,64){
        rep(j,1,i){
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
        }
    }

    rep(x,0,64){
        rep(y,0,64){
            rep(i,0,x){
                rep(j,0,y){
                    if((i+2*j)%3==0) {
                        dp[x][y]=(dp[x][y]+C[x][i]*C[y][j])%mod;
                        //if(x==1&&y==1) printf("now  dp[x][y]=%lld i = %d j = %d\n",dp[x][y],i,j);
                    }
                }
            }
        }
    }
}
int main(){
    init();
    int T,cas=0; scanf("%d",&T);
    while(T--){
        scanf("%lld%lld",&n,&a);
        n1=n2=0; int cou=1;
        while(a){
            if(a&1){
                if(cou%2) n1++;
                else n2++;
            }
            cou++;
            a/=2;
        }
        ll ans=0;
        rep(i,0,n1){
            rep(j,0,n2){
                ll tmp=((n1+n2-i-j)%2==1?-1:1);
                ans=(ans+tmp*C[n1][i]*C[n2][j]%mod*quick(dp[i][j],n)%mod+mod)%mod;
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}