2017多校(5) A、Rikka with Candies bitset

Rikka with Candies //bitset

bitset a、b分别保存a、b数组出现与否,a左移和右移就可以表示a(+/-)k。
把b的倍数在b1中做出来。

#pragma GCC optimize(2)
#include
#include
#include
using namespace std;
const int maxn=5e4+5;
bitseta,b,b1,ans;
int n,m,q;
int read()
{
    int ans=0;
    char ch;
    ch=getchar();
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9')
    {
        ans=ans*10+ch-'0';
        ch=getchar();
    }
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        a.reset();
        b.reset();
        b1.reset();
        ans.reset();
        n=read(),m=read(),q=read();
        int x;
        int upper=0;
        for(int i=1;i<=n;i++)
        {
            x=read();
            a.set(x);
        }
        for(int i=1;i<=m;i++)
        {
            x=read();
            b.set(x);
            upper=max(upper,x);
        }
        for(int i=upper;i>=0;i--)//倒序枚举因为小的数的倍数会影响到大的数在b1中的数值
        {
            ans[i]=(b1&(a>>i)).count()&1;//统计答案
            if(b[i]) for(int j=0;j<=upper;j+=i) b1.flip(j);
        }
        while(q--)
        {
            x=read();
            if(ans[x]) printf("1\n");
            else printf("0\n");
        }
    }
    return 0;
}

你可能感兴趣的:(数论,杂题)