/tmp/solutions/build/range_reverse_range_sum-main.cpp:
1| |#include <common.h>
2| |prelude;
3| |
4| |namespace {
5| |
6| |constexpr int N = 200003;
7| |
8| |struct Node {
9| | int val;
10| | u64 sum;
11| | int rev;
12| | int size;
13| | int l, r;
14| |} node[N];
15| |
16| 151k|def maintain(int x) -> void {
17| 151k| node[x].size = node[node[x].l].size + 1 + node[node[x].r].size;
18| 151k| node[x].sum = node[node[x].l].sum + node[x].val + node[node[x].r].sum;
19| 151k|}
20| |
21| 303k|def build(int l, int r) -> int {
22| 303k| if (l > r) return 0;
^151k
------------------
| Branch (22:7): [True: 50.00%, False: 50.00%]
------------------
23| 151k| int m = (l + r) / 2;
24| 151k| node[m].l = build(l, m - 1);
25| 151k| node[m].r = build(m + 1, r);
26| 151k| maintain(m);
27| 151k| return m;
28| 303k|}
29| |
30| 31|def pushdown(int x) -> void {
31| 31| if (node[x].rev) {
------------------
| Branch (31:7): [True: 0.00%, False: 100.00%]
------------------
32| 0| std::swap(node[x].l, node[x].r);
33| 0| node[node[x].l].rev ^= 1;
34| 0| node[node[x].r].rev ^= 1;
35| 0| node[x].rev = 0;
36| 0| }
37| 31|}
38| |
39| 1|def reverse(int x) -> void {
40| 1| if (x == 0) return;
^0
------------------
| Branch (40:7): [True: 0.00%, False: 100.00%]
------------------
41| 1| node[x].rev ^= 1;
42| 1|}
43| |
44| 16|def zig(int &x) -> void {
45| 16| int l = node[x].l;
46| 16| node[x].l = node[l].r;
47| 16| maintain(x);
48| 16| node[l].r = x;
49| 16| x = l;
50| 16|}
51| |
52| 13|def zag(int &x) -> void {
53| 13| int r = node[x].r;
54| 13| node[x].r = node[r].l;
55| 13| maintain(x);
56| 13| node[r].l = x;
57| 13| x = r;
58| 13|}
59| |
60| 16|def splay(int &x, int k) -> void {
61| 16| pushdown(x);
62| 16| if (int &l = node[x].l, &r = node[x].r, size = node[l].size; k == size) {
------------------
| Branch (62:64): [True: 6.25%, False: 93.75%]
------------------
63| 1| return;
64| 15| } else if (k < size) {
------------------
| Branch (64:14): [True: 40.00%, False: 60.00%]
------------------
65| 6| pushdown(l);
66| 6| if (int &ll = node[l].l, &lr = node[l].r, size = node[ll].size; k == size) {
------------------
| Branch (66:69): [True: 0.00%, False: 100.00%]
------------------
67| 0| zig(x);
68| 6| } else if (k < size) {
------------------
| Branch (68:16): [True: 50.00%, False: 50.00%]
------------------
69| 3| splay(ll, k);
70| 3| zig(x);
71| 3| zig(x);
72| 3| } else {
73| 3| splay(lr, k - size - 1);
74| 3| zag(l);
75| 3| zig(x);
76| 3| }
77| 9| } else {
78| 9| pushdown(r);
79| 9| k -= size + 1;
80| 9| if (int &rl = node[r].l, &rr = node[r].r, size = node[rl].size; k == size) {
------------------
| Branch (80:69): [True: 11.11%, False: 88.89%]
------------------
81| 1| zag(x);
82| 8| } else if (k < size) {
------------------
| Branch (82:16): [True: 87.50%, False: 12.50%]
------------------
83| 7| splay(rl, k);
84| 7| zig(r);
85| 7| zag(x);
86| 7| } else {
87| 1| splay(rr, k - size - 1);
88| 1| zag(x);
89| 1| zag(x);
90| 1| }
91| 9| }
92| 16|}
93| |
94| |} // namespace
95| |
96| 1|int main() {
97| 1| rd rd;
98| 1| wt wt;
99| 1| int n = rd.uh();
100| 1| int q = rd.uh();
101| 1|#ifdef LOCAL
102| 1| std::memset(node, 0, sizeof(Node) * (n + 3));
103| 1|#endif
104| 151k| for (int i = 2; i <= n + 1; ++i) node[i].val = rd.uw();
^151k^151k
------------------
| Branch (104:19): [True: 100.00%, False: 0.00%]
------------------
105| 1| int x = build(1, n + 2);
106| 1| if (n == 0) rd.skip(1);
^0
------------------
| Branch (106:7): [True: 0.00%, False: 100.00%]
------------------
107| 2| while (q--) {
------------------
| Branch (107:10): [True: 50.00%, False: 50.00%]
------------------
108| 1| int t = rd.u1();
109| 1| int l = rd.uh();
110| 1| int r = rd.uh();
111| 1| splay(x, l);
112| 1| splay(node[x].r, r - l);
113| 1| if (t == 0) {
------------------
| Branch (113:9): [True: 100.00%, False: 0.00%]
------------------
114| 1| reverse(node[node[x].r].l);
115| 1| }
116| 1| if (t == 1) {
------------------
| Branch (116:9): [True: 0.00%, False: 100.00%]
------------------
117| 0| wt.ud(node[node[node[x].r].l].sum);
118| 0| }
119| 1| }
120| 1| return 0;
121| 1|}