CSU 1911 Card Game 快速沃尔什变换（FWT）模板题

最新推荐文章于 2020-09-23 20:39:58 发布

天使v之翼

最新推荐文章于 2020-09-23 20:39:58 发布

阅读量1.3k

点赞数 1

分类专栏：数论文章标签： CSU 1911 快速沃尔什变换 FWT

本文为博主原创文章，未经博主允许不得转载。

本文链接：https://blog.csdn.net/xuanandting/article/details/70991387

版权

数论专栏收录该内容

15 篇文章 0 订阅

订阅专栏

题意：给你两个二进制数的集合，给出q次询问，输出两个集合之间元素或的值等于查询值的种数。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define LL long long
const int maxn = 1<<18;
LL a[maxn],b[maxn];
int n,m;
char s[20];

void FWT(LL a[],int n)
{
    for(int d=1;d<n;d<<=1)
        for(int m=d<<1,i=0;i<n;i+=m)
            for(int j=0;j<d;j++)
            {
                int x=a[i+j],y=a[i+j+d];
                a[i+j+d]=x+y;
            }
}

void UFWT(LL a[],int n)
{
    for(int d=1;d<n;d<<=1)
        for(int m=d<<1,i=0;i<n;i+=m)
            for(int j=0;j<d;j++)
            {
                int x=a[i+j],y=a[i+j+d];
                a[i+j+d]=y-x;
            }
}
void solve(LL a[],LL b[],int n)
{
    FWT(a,n);
    FWT(b,n);
    for(int i=0;i<n;i++) a[i]=a[i]*b[i];
    UFWT(a,n);
}

int main()
{
    int t,q;
    scanf("%d",&t);
    for(int cas=1;cas<=t;cas++)
    {
        memset(a,0,sizeof a);
        memset(b,0,sizeof b);
        scanf("%d%d",&n,&m);
        int l,x;
        for(int i=0;i<n;i++)
        {
            x=0;
            scanf("%s",s);
            l=strlen(s);
            for(int j=0;j<l;j++) x=(x<<1)+s[j]-'0';
            a[x]++;
        }
        for(int i=0;i<n;i++)
        {
            x=0;
            scanf("%s",s);
            l=strlen(s);
            for(int j=0;j<l;j++) x=(x<<1)+s[j]-'0';
            b[x]++;
        }
        solve(a,b,(1<<m));
        scanf("%d",&q);
        printf("Case #%d:\n",cas);
        while(q--)
        {
            scanf("%s",s);
            x=0;
            l=strlen(s);
            for(int i=0;i<l;i++) x=(x<<1)+s[i]-'0';
            printf("%lld\n",a[x]);
        }
    }
    return 0;
}