bzoj 4810: [Ynoi2017]由乃的玉米田 莫队算法+bitset

题意

给你一个序列a，长度为n，有m次操作，每次询问一个区间是否可以选出两个数它们的差为x，或者询问一个区间是否可以选出两个数它们的和为x，或者询问一个区间是否可以选出两个数它们的乘积为x ，这三个操作分别为操作1,2,3选出的这两个数可以是同一个位置的数
定义c为每次的x和ai中的最大值，ai >= 0，每次的x>=2，n,m,c <= 100000

分析

考虑莫队，维护一个桶和两个bitset，一个正的一个反的，对于询问把bitset各种移然后&起来即可。询问乘积的话就暴力枚举因子就好了。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>#include<bitset>#include<cmath>using namespace std;const int N=100005;int n,m,bel[N],block,t[N],a[N],mx;bitset<N> g,f,rev;struct query{int id,l,r,op,x,ans;}q[N];bool cmp(query a,query b){    return bel[a.l]<bel[b.l]||bel[a.l]==bel[b.l]&&a.r<b.r;}bool cmpid(query a,query b){    return a.id<b.id;}void updata(int x,int op){    if (op==1)    {        if (!t[x]) f[x]=1,rev[mx-x]=1;        t[x]++;    }    else    {        t[x]--;        if (!t[x]) f[x]=0,rev[mx-x]=0;    }}int check(int op,int x){    if (op==1)    {        g=f;        g>>=x;        g&=f;        if (g.count()) return 1;        else return 0;    }    else if (op==2)    {        g=rev;        g>>=(mx-x);        g&=f;        if (g.count()) return 1;        else return 0;    }    else    {        int w=sqrt(x);        for (int i=1;i<=w;i++)            if (x%i==0&&f[i]&&f[x/i]) return 1;        return 0;    }}void solve(){    for (int i=1,l=1,r=0;i<=m;i++)    {        for (;r<q[i].r;r++) updata(a[r+1],1);        for (;l>q[i].l;l--) updata(a[l-1],1);        for (;r>q[i].r;r--) updata(a[r],-1);        for (;l<q[i].l;l++) updata(a[l],-1);        q[i].ans=check(q[i].op,q[i].x);    }}int main(){    scanf("%d%d",&n,&m);    block=sqrt(n);    for (int i=1;i<=n;i++) scanf("%d",&a[i]),bel[i]=(i+block-1)/block,mx=max(mx,a[i]);    for (int i=1;i<=m;i++)    {        scanf("%d%d%d%d",&q[i].op,&q[i].l,&q[i].r,&q[i].x);        mx=max(mx,q[i].x);        q[i].id=i;    }    sort(q+1,q+m+1,cmp);    solve();    sort(q+1,q+m+1,cmpid);    for (int i=1;i<=m;i++)        if (q[i].ans) printf("yuno\n");        else printf("yumi\n");    return 0;}