[CSU1911]Card Game

vjudge

题意

两个数组\(\{a_i\}\)和\(\{b_i\}\)，求从中分别选出两个数或运算结果为\(x\)的方案数。

sol

裸的FWT。

code

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int gi(){
    int x=0,w=1;char ch=getchar();
    while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
    if (ch=='-') w=0,ch=getchar();
    while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return w?x:-x;
}
#define ll long long
const int N = 1<<18;
int T,n,m,q,len;ll a[N],b[N];
int read(){
    char s[30];scanf("%s",s);int x=0;
    for (int i=0,l=strlen(s);i<l;++i) x=(x<<1)+s[i]-'0';
    return x;
}
void fwt(ll *P,int len,int opt){
    for (int i=1;i<len;i<<=1)
        for (int p=i<<1,j=0;j<len;j+=p)
            for (int k=0;k<i;++k)
                P[j+k+i]+=P[j+k]*opt;
}
void mul(ll *a,ll *b,int len){
    for (int i=0;i<len;++i) a[i]=a[i]*b[i];
}
int main(){
    T=gi();
    for (int zsy=1;zsy<=T;++zsy){
        printf("Case #%d:\n",zsy);
        n=gi();m=gi();len=1<<m;
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for (int i=1;i<=n;++i) ++a[read()];
        for (int i=1;i<=n;++i) ++b[read()];
        fwt(a,len,1);fwt(b,len,1);mul(a,b,len);
        fwt(a,len,-1);
        q=gi();while (q--) printf("%lld\n",a[read()]);
    }
    return 0;
}