BZOJ4810:[YNOI2017]由乃的玉米田(莫队,bitset)

Description

由乃在自己的农田边散步，她突然发现田里的一排玉米非常的不美。这排玉米一共有N株，它们的高度参差不齐。

由乃认为玉米田不美，所以她决定出个数据结构题

 

这个题是这样的：

给你一个序列a，长度为n，有m次操作，每次询问一个区间是否可以选出两个数它们的差为x，或者询问一个区间是

否可以选出两个数它们的和为x，或者询问一个区间是否可以选出两个数它们的乘积为x ，这三个操作分别为操作1

,2,3选出的这两个数可以是同一个位置的数

Input

第一行两个数n,m

后面一行n个数表示ai

后面m行每行四个数opt l r x

opt表示这个是第几种操作，l,r表示操作的区间，x表示这次操作的x

定义c为每次的x和ai中的最大值，ai >= 0，每次的x>=2n,m,c <= 100000

Output

对于每个询问，如果可以，输出yuno，否则输出yumi

Sample Input

5 5
1 1 2 3 4
2 1 1 2
1 1 2 2
3 1 1 1
3 5 5 16
1 2 3 4

Sample Output

yuno
yumi
yuno
yuno
yumi

Solution

先莫个队，然后对值域开个$bitset$。

差相等就是$f$并上$f$右移$x$不为$0$。

和相等就是$f$并上翻转的$f$右移$N-x$位不为$0$。

积的话就$sqrt$枚举$x$的的因子然后查询存在性就好了。

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<bitset>
#include<cmath>
#include<algorithm>
#define N (100000)
using namespace std;

struct Que{int opt,l,r,x,id;}Q[N+1];
int n,m,unit,opt,l,r,x,a[N+1],ID[N+1];
int ans[N+9],Keg[N+9];
bitset<N+1>f,g;

inline int read()
{
    int x=0,w=1; char c=getchar();
    while (!isdigit(c)) {if (c=='-') w=-1; c=getchar();}
    while (isdigit(c)) x=x*10+c-'0', c=getchar();
    return x*w;
    
}

void Ins(int p)
{
    if (!Keg[a[p]]) f[a[p]]=1, g[N-a[p]]=1;
    ++Keg[a[p]];
}

void Del(int p)
{
    --Keg[a[p]];
    if (!Keg[a[p]]) f[a[p]]=0, g[N-a[p]]=0;
}

bool check(int opt,int x)
{
    if (opt==1) return (f&(f>>x)).any();
    if (opt==2) return (f&(g>>(N-x))).any();
    if (opt==3)
    {
        for (int i=1; i<=sqrt(x); ++i)
            if (x%i==0 && f[i] && f[x/i]) return 1;
        return 0;
    }
}

bool cmp(Que a,Que b)
{
    if (ID[a.l]==ID[b.l]) return a.r<b.r;
    return ID[a.l]<ID[b.l];
}

int main()
{
    n=read(); m=read(); unit=sqrt(n);
    for (int i=1; i<=n; ++i) ID[i]=i/unit;
    for (int i=1; i<=n; ++i) a[i]=read();
    for (int i=1; i<=m; ++i)
    {
        opt=read(); l=read(); r=read(); x=read();
        Q[i]=(Que){opt,l,r,x,i};
    }
    sort(Q+1,Q+m+1,cmp);
    int l=1,r=0;
    for (int i=1; i<=m; ++i)
    {
        while (l<Q[i].l) Del(l++);
        while (l>Q[i].l) Ins(--l);
        while (r<Q[i].r) Ins(++r);
        while (r>Q[i].r) Del(r--);
        ans[Q[i].id]=check(Q[i].opt,Q[i].x);
    }
    for (int i=1; i<=m; ++i) puts(ans[i]?"yuno":"yumi");
}