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

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdio>
 4 #include<bitset>
 5 #include<cmath>
 6 #include<algorithm>
 7 #define N (100000)
 8 using namespace std;
 9 
10 struct Que{int opt,l,r,x,id;}Q[N+1];
11 int n,m,unit,opt,l,r,x,a[N+1],ID[N+1];
12 int ans[N+9],Keg[N+9];
13 bitset<N+1>f,g;
14 
15 inline int read()
16 {
17     int x=0,w=1; char c=getchar();
18     while (!isdigit(c)) {if (c=='-') w=-1; c=getchar();}
19     while (isdigit(c)) x=x*10+c-'0', c=getchar();
20     return x*w;
21     
22 }
23 
24 void Ins(int p)
25 {
26     if (!Keg[a[p]]) f[a[p]]=1, g[N-a[p]]=1;
27     ++Keg[a[p]];
28 }
29 
30 void Del(int p)
31 {
32     --Keg[a[p]];
33     if (!Keg[a[p]]) f[a[p]]=0, g[N-a[p]]=0;
34 }
35 
36 bool check(int opt,int x)
37 {
38     if (opt==1) return (f&(f>>x)).any();
39     if (opt==2) return (f&(g>>(N-x))).any();
40     if (opt==3)
41     {
42         for (int i=1; i<=sqrt(x); ++i)
43             if (x%i==0 && f[i] && f[x/i]) return 1;
44         return 0;
45     }
46 }
47 
48 bool cmp(Que a,Que b)
49 {
50     if (ID[a.l]==ID[b.l]) return a.r<b.r;
51     return ID[a.l]<ID[b.l];
52 }
53 
54 int main()
55 {
56     n=read(); m=read(); unit=sqrt(n);
57     for (int i=1; i<=n; ++i) ID[i]=i/unit;
58     for (int i=1; i<=n; ++i) a[i]=read();
59     for (int i=1; i<=m; ++i)
60     {
61         opt=read(); l=read(); r=read(); x=read();
62         Q[i]=(Que){opt,l,r,x,i};
63     }
64     sort(Q+1,Q+m+1,cmp);
65     int l=1,r=0;
66     for (int i=1; i<=m; ++i)
67     {
68         while (l<Q[i].l) Del(l++);
69         while (l>Q[i].l) Ins(--l);
70         while (r<Q[i].r) Ins(++r);
71         while (r>Q[i].r) Del(r--);
72         ans[Q[i].id]=check(Q[i].opt,Q[i].x);
73     }
74     for (int i=1; i<=m; ++i) puts(ans[i]?"yuno":"yumi");
75 }