[Zjoi 2017] bzoj4785 树状数组 [二维线段树]

Description:
两个操作:
$1$.区间等概率选一个数$+1$。
$2$.询问倒序树状数组正确的概率。

---

Solution:
实际上等价于求$a_{l-1}=a_r$的概率。
利用二维线段树保存两位相同的概率，每次更新对应区间，标记永久化即可。
注意$l=1$时即为计算$sum$ $xor$ $a[r]=0$的概率，由于每次修改都会改变$sum$，维护一个奇偶性即可。

---

#include <bits/stdc++.h>
using namespace std;
const int maxn = 3e5 + 5, SZ = 4e7 + 5, P = 998244353;
int n, m, cnt, p, ans;
int rt[maxn * 4], lc[SZ], rc[SZ],  data[SZ];
int read() {
    int x = 0;
    char c = getchar();
    while(!isdigit(c)) {
        c = getchar();
    }
    while(isdigit(c)) {
        x = x * 10 + c - '0';
        c = getchar();
    }
    return x;
}
inline int merge(int x, int y) {
    return (1LL * x * (1 - y + P) % P + 1LL * y * (1 - x + P) % P) % P;
}
int power(int x, int t) {
    int ret = 1;
    for(; t; t >>= 1, x = 1LL * x * x % P) {
        if(t & 1) {
            ret = 1LL * ret * x % P;
        }
    }
    return ret;
}
void update_2(int l, int r, int &x, int a, int b, int t) {
    if(l > b || r < a) {
        return;
    }
    if(!x) {
        x = ++cnt;
        data[x] = 1;
    }
    if(l >= a && r <= b) {
        data[x] = merge(data[x], t);
        return;
    }
    int mid = (l + r) >> 1;
    update_2(l, mid, lc[x], a, b, t);
    update_2(mid + 1, r, rc[x], a, b, t);
}
void update_1(int l, int r, int x, int a, int b, int A, int B, int t) {
    if(l > b || r < a) {
        return;
    }
    if(l >= a && r <= b) {
        update_2(0, n, rt[x], A, B, t);
        return;
    }
    int mid = (l + r) >> 1;
    update_1(l, mid, x << 1, a, b, A, B, t);
    update_1(mid + 1, r, x << 1 | 1, a, b, A, B, t);
}
void query_2(int l, int r, int x, int p) {
    if(!x) {
        return;
    }
    ans = merge(ans, (1 - data[x] + P) % P);
    if(l == r) {
        return;
    }
    int mid = (l + r) >> 1;
    if(p <= mid) {
        query_2(l, mid, lc[x], p);
    } else {
        query_2(mid + 1, r, rc[x], p);
    }
}
void query_1(int l, int r, int x, int p1, int p2) {
    query_2(0, n, rt[x], p2);
    if(l == r) {
        return;
    } 
    int mid = (l + r) >> 1;
    if(p1 <= mid) {
        query_1(l, mid, x << 1, p1, p2);
    } else {
        query_1(mid + 1, r, x << 1 | 1, p1, p2);
    }
}
int main() {
    n = read();
    m = read(); 
    while(m--) {
        int opt = read(), l = read(), r = read();
        if(opt == 1) {
            int x = power(r - l + 1, P - 2);
            update_1(0, n, 1, 0, l - 1, l, r, x);
            update_1(0, n, 1, l, r, r + 1, n, x);
            x = x * 2 % P;
            update_1(0, n, 1, l, r, l, r, x);
            ++p;
        } else {
            ans = 1;
            query_1(0, n, 1, l - 1, r);
            if(l == 1 && (p & 1)) {
                ans = (P + 1 - ans) % P;
            }
            printf("%d\n", ans);
        }
    }
    return 0;
}