bzoj 4810 由乃的玉米田 - bitset - 莫队算法
由乃在自己的农田边散步,她突然发现田里的一排玉米非常的不美。这排玉米一共有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
题目大意 (数据结构裸题不需要大意)。(这是由乃OI?)
显然bitset(别问我怎么知道的)。
然而考虑分块,MLE?所以果断否决掉。
是有个基于分块有很省内存的算法?莫队啊,我们只需要2个bitset和一个cnt数组就好了。
现在来考虑具体的操作。
1)区间内是否存在两个数差为x。这个很简单,bitset一个基本的应用。bitset维护一个值域,然后将所有数都减去x(相当于将这个bitset右移x位),再看有没有数相等(新bitset按位与旧bitset得到的bitset是否存在某一位为1,调用它的函数any()就好了)
2)区间内是否存在两个数和为x。考虑a + b = x,加法不是很好处理,就转化成减法,得到a - (-b) = x。因为bitset不支持负数下标,把大小增大常数又变大,所以考虑用2个bitset。新的bitset维护某一个上限加上-b后的值域。查询的时候我们希望将第一个bitset右移x位,显然负数位才有价值,但是右移溢出的比特位会被直接舍去。所以考虑向左移,把右移x后,没有溢出的部分全部删去。这样的话就左移(limit - x)位,在和第二个bitset进行按位与,然后判断是否存在某一位为1。
3)还是bitset?想多了。。。Tag害死人啊。。值域只有1e5。是不是直接根号大暴力(枚举因子)就完事了?(然后我忘判因子了,就WA了几次)
Code
1 /** 2 * bzoj 3 * Problem#4810 4 * Accepted 5 * Time:13672ms 6 * Memory:4472k 7 */ 8 #include<bits/stdc++.h> 9 using namespace std; 10 typedef bool boolean; 11 12 typedef class Query { 13 public: 14 int opt; 15 int l; 16 int r; 17 int x; 18 int lid; 19 int id; 20 21 Query():opt(opt), l(l), r(r), x(x), lid(lid) { } 22 23 boolean operator < (Query b) const { 24 if(lid != b.lid) return lid < b.lid; 25 return r < b.r; 26 } 27 }Query; 28 29 #define limit 100001 30 int n, m, cs; 31 int *arr; 32 bitset<100005> s, s1, rs; 33 Query* qs; 34 inline void init() { 35 scanf("%d%d", &n, &m); 36 arr = new int[(n + 1)]; 37 qs = new Query[(m + 1)]; 38 cs = sqrt(n + 0.5); 39 for(int i = 1; i <= n; i++) 40 scanf("%d", arr + i); 41 for(int i = 1; i <= m; i++) { 42 scanf("%d%d%d%d", &qs[i].opt, &qs[i].l, &qs[i].r, &qs[i].x); 43 qs[i].lid = qs[i].l / cs, qs[i].id = i; 44 } 45 } 46 47 boolean *res; 48 int cnt[100005]; 49 inline void extends(int pos, int val) { 50 cnt[arr[pos]] += val; 51 if(cnt[arr[pos]] == 1) s[arr[pos]] = 1, rs[limit - arr[pos]] = 1; 52 if(cnt[arr[pos]] == 0) s[arr[pos]] = 0, rs[limit - arr[pos]] = 0; 53 } 54 55 inline void solve() { 56 res = new boolean[(m + 1)]; 57 sort(qs + 1, qs + m + 1); 58 int mdzzl = 1, mdzzr = 0; 59 boolean aFlag = false; 60 for(int i = 1; i <= m; i++) { 61 while(mdzzr < qs[i].r) extends(++mdzzr, 1); 62 while(mdzzr > qs[i].r) extends(mdzzr--, -1); 63 while(mdzzl > qs[i].l) extends(--mdzzl, 1); 64 while(mdzzl < qs[i].l) extends(mdzzl++, -1); 65 switch(qs[i].opt) { 66 case 1: 67 s1 = (s << qs[i].x) & s; 68 res[qs[i].id] = s1.any(); 69 break; 70 case 2: 71 s1 = (s << (limit - qs[i].x)) & rs; 72 res[qs[i].id] = s1.any(); 73 break; 74 case 3: 75 aFlag = false; 76 for(int j = 1; j * j <= qs[i].x; j++) 77 if((qs[i].x % j == 0) && s[j] && s[qs[i].x / j]) { 78 aFlag = true; 79 break; 80 } 81 res[qs[i].id] = aFlag; 82 break; 83 default: 84 break; 85 } 86 } 87 for(int i = 1; i <= m; i++) 88 puts((res[i]) ? ("yuno") : ("yumi")); 89 } 90 91 int main() { 92 init(); 93 solve(); 94 return 0; 95 }