hdu 6085(bitset优化)

来源:互联网 发布:如何考环评工程师 知乎 编辑:程序博客网 时间:2024/06/10 09:42


题意:

 存在两个长度分别为 n,m 的数组 A,B 。有 q 个询问,每个询问给出一个数字 k ,可以得到使得 AimodBj=k 的种数。求该种数的奇偶性。

思路:对A数组建立一个bitset,然后对于B数组,对它的倍数建立一个bitset,当枚举i为余数的时候,大于i的B[j]的倍数都存放在了第二个bitset里面了。因为这里讨论的奇偶性,所以当一个地方出现两个数的时候,直接flip。

用bitset优化的复杂度为O(n^2/32)

#include<bits/stdc++.h>using namespace std;typedef long long ll;typedef pair<int,int> P;#define fi first#define se second#define INF 0x3f3f3f3f#define clr(x,y) memset(x,y,sizeof x)#define PI acos(-1.0)#define ITER set<int>::iteratorconst int Mod = 1e9 + 7;const int maxn = 5e4 + 10;int a[maxn],b[maxn];bitset<maxn>B1,B2;int ans[maxn];int main(){    int Tcase;scanf("%d",&Tcase);    while( Tcase --)    {        B1.reset(),B2.reset();clr(ans,0);        int n,m,k;scanf("%d%d%d",&n,&m,&k);        for(int i = 1; i <= n; i ++)scanf("%d",&a[i]),B1.set(a[i]);        for(int i = 1; i <= m; i ++)scanf("%d",&b[i]);sort(b + 1,b + m + 1);        int pos = m;        for(int i = b[m];i >= 0; i --)        {            if(b[pos] > i && pos >= 1)            {                for(int j = 0;j <= b[m]; j += b[pos])B2.flip(j);pos --;            }            ans[i] ^= ((B2 << i) & B1).count() & 1;        }        for(int i = 1; i <= k;i ++){int x;scanf("%d",&x);printf("%d\n",ans[x]);}    }    return 0;}