块状数组
建立块状数组¶
块状数组,即把一个数组分为几个块,块内信息整体保存,若查询时遇到两边不完整的块直接暴力查询。一般情况下,块的长度为 O(\sqrt{n}) 。详细分析可以阅读 2017 年国家集训队论文中徐明宽的《非常规大小分块算法初探》。
下面直接给出一种建立块状数组的代码。
1 2 3 4 5 6 7 8 9 10 | num = sqrt(n); for (int i = 1; i <= num; i++) st[i] = n / num * (i - 1) + 1, ed[i] = n / num * i; ed[num] = n; for (int i = 1; i <= num; i++) { for (int j = st[i]; j <= ed[i]; j++) { belong[j] = i; } size[i] = ed[i] - st[i] + 1; } |
其中 st[i] ed[i]
为块的起点和终点, size[i]
为块的大小。
保存与修改块内信息¶
例题 1: 教主的魔法¶
我们要询问一个块内大于等于一个数的数的个数,所以需要一个 t
数组对块内排序。对于整块的修改,使用类似于标记永久化的方式保存。时间复杂度 O(n\sqrt{n}\log n)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | void Sort(int k) { for (int i = st[k]; i <= ed[k]; i++) t[i] = a[i]; sort(t + st[k], t + ed[k] + 1); } void Modify(int l, int r, int c) { int x = belong[l], y = belong[r]; if (x == y) { for (int i = l; i <= r; i++) a[i] += c; Sort(x); return; } for (int i = l; i <= ed[x]; i++) a[i] += c; for (int i = st[y]; i <= r; i++) a[i] += c; for (int i = x + 1; i < y; i++) dlt[i] += c; Sort(x); Sort(y); } int Answer(int l, int r, int c) { int ans = 0, x = belong[l], y = belong[r]; if (x == y) { for (int i = l; i <= r; i++) if (a[i] + dlt[x] >= c) ans++; return ans; } for (int i = l; i <= ed[x]; i++) if (a[i] + dlt[x] >= c) ans++; for (int i = st[y]; i <= r; i++) if (a[i] + dlt[y] >= c) ans++; for (int i = x + 1; i <= y - 1; i++) ans += ed[i] - (lower_bound(t + st[i], t + ed[i] + 1, c - dlt[i]) - t) + 1; return ans; } |
例题 2:寒夜方舟¶
两种操作:
- 区间 [x,y] 每个数都变成 z
- 查询区间 [x,y] 内小于等于 z 的数的个数
用 dlt
保存现在块内是否被整体赋值了。用一个值表示没有。对于边角块,查询前要 pushdown
,把块内存的信息下放到每一个数上。赋值之后记得重新 sort
一遍。其他方面同上题。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | void Sort(int k) { for (int i = st[k]; i <= ed[k]; i++) t[i] = a[i]; sort(t + st[k], t + ed[k] + 1); } void PushDown(int x) { if (dlt[x] != 0x3f3f3f3f3f3f3f3fll) for (int i = st[x]; i <= ed[x]; i++) a[i] = t[i] = dlt[x]; dlt[x] = 0x3f3f3f3f3f3f3f3fll; } void Modify(int l, int r, int c) { int x = belong[l], y = belong[r]; PushDown(x); if (x == y) { for (int i = l; i <= r; i++) a[i] = c; Sort(x); return; } PushDown(y); for (int i = l; i <= ed[x]; i++) a[i] = c; for (int i = st[y]; i <= r; i++) a[i] = c; Sort(x); Sort(y); for (int i = x + 1; i < y; i++) dlt[i] = c; } int Binary_Search(int l, int r, int c) { int ans = l - 1, mid; while (l <= r) { mid = (l + r) / 2; if (t[mid] <= c) ans = mid, l = mid + 1; else r = mid - 1; } return ans; } int Answer(int l, int r, int c) { int ans = 0, x = belong[l], y = belong[r]; PushDown(x); if (x == y) { for (int i = l; i <= r; i++) if (a[i] <= c) ans++; return ans; } PushDown(y); for (int i = l; i <= ed[x]; i++) if (a[i] <= c) ans++; for (int i = st[y]; i <= r; i++) if (a[i] <= c) ans++; for (int i = x + 1; i <= y - 1; i++) { if (0x3f3f3f3f3f3f3f3fll == dlt[i]) ans += Binary_Search(st[i], ed[i], c) - st[i] + 1; else if (dlt[i] <= c) ans += size[i]; } return ans; } |
练习¶
build本页面最近更新:,更新历史
edit发现错误?想一起完善? 在 GitHub 上编辑此页!
people本页面贡献者:OI-wiki
copyright本页面的全部内容在 CC BY-SA 4.0 和 SATA 协议之条款下提供,附加条款亦可能应用