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
,block,t
,a
,mx;
bitset<N> g,f,rev;
struct query{int id,l,r,op,x,ans;}q
;

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;
}