Avito Code Challenge 2018 G. Magic multisets（线段树）

题目链接

题意：
$\ \ \ \ $维护一个多重集合，两种操作：
1 给区间$[l, r]$的多重集合添加一个数$x$，如果有集合里面已经存在$x$，那么该集合里面的所有元素翻倍，否则插入该元素。
2 求区间$[l, r]$的集合的大小和。

思路：
$\ \ \ \ $对于添加操作，默认先给$[l, r]$乘以$2$， 然后暴力$x$当前覆盖的所有区间，没有覆盖的区间和$[l,r]$求下交，给这段区间乘以$2$的逆元再$+1$，然后将可以合并的区间合并。不知道为啥暴力这部分的时间复杂度$O(\log n)$ …
$\ \ \ \ $ 然后就只需要维护区间加法乘法了。可以维护这样的形式：$a \cdot x+b$，假如当前位置的和表示形式是$a \cdot sum + b$，当一个区间乘以$d$的时候，形式变成了$d \cdot(a \cdot sum + b) = (a \cdot d) \cdot sum + (d \cdot b)$, 同样加法的话变成$a \cdot sum + b + d = a \cdot sum + (d + b)$，懒惰标记延迟下传$a$和$b$就行了。

#include<bits/stdc++.h>
typedef long long ll;
const int maxn = 2e5 + 10;
using namespace std;

typedef pair<int, int> pa;
ll sum[maxn * 4], add[maxn * 4], mul[maxn * 4];
ll n, m, mod;
set<pa> st[maxn];

void build(int o, int l, int r) {
    mul[o] = 1; add[o] = sum[o] = 0;
    if(l == r) return ;
    int mid = (l + r) >> 1;
    build(o << 1, l, mid);
    build(o << 1 | 1, mid + 1, r);
}

void push_down(int o, int l, int r) {
    int o1 = o << 1, o2 = o1 | 1;
    if(mul[o] != 1) {
       mul[o1] = mul[o1] * mul[o] % mod;
       mul[o2] = mul[o2] * mul[o] % mod;
       add[o1] = (add[o1] * mul[o]) % mod;
       add[o2] = (add[o2] * mul[o]) % mod;
       mul[o] = 1; 
    }
    if(add[o]) {
        add[o1] = (add[o1] + add[o]) % mod;
        add[o2] = (add[o2] + add[o]) % mod;
        add[o] = 0;
    }
}

void push_up(int o, int l, int r) {
    int mid = (l + r) >> 1;
    int o1 = o << 1, o2 = o1 | 1;
    sum[o] = (mul[o1] * sum[o1] + add[o1] * (mid - l + 1)) % mod;
    sum[o] = (sum[o] + mul[o2] * sum[o2] + add[o2] * (r - mid)) % mod;
}

void update(int o, int l, int r, int ql, int qr, ll x, int flag) {
    if(l > qr || r < ql) return ;
    if(l >= ql && r <= qr) {
        if(flag == 1) {
            mul[o] = mul[o] * x % mod;
            add[o] = add[o] * x % mod;
        } else {
            add[o] = (add[o] + x) % mod;
        }
        return ;
    }
    int mid = (l + r) >> 1;
    push_down(o, l, r);
    update(o << 1, l, mid, ql, qr, x, flag);
    update(o << 1 | 1, mid + 1, r, ql, qr, x, flag);
    push_up(o, l, r);
}

ll query(int o, int l, int r, int ql, int qr) {
    if(l > qr || r < ql) return 0;
    if(l >= ql && r <= qr) return (sum[o] * mul[o] + add[o] * (r - l + 1)) % mod;
    int mid = (l + r) >> 1;
    push_down(o, l, r);
    ll p1 = query(o << 1, l, mid, ql, qr);
    ll p2 = query(o << 1 | 1, mid + 1, r, ql, qr);
    push_up(o, l, r);
    return (p1 + p2) % mod; 
}

void merge(int x, int L, int R) {
    set<pa>::iterator it;
    it = st[x].lower_bound(pa(L, L));
    int mergeL = L, mergeR = R;
    for( ; it != st[x].end(); it++) {
        set<pa>::iterator las = it; las--;
        int l = (*las).second + 1;
        int r = (*it).first - 1;
        int upl = max(l, L);
        int upr = min(r, R);
        if(upr >= upl) update(1, 1, n, upl, upr, (mod + 1) / 2, 1);
        if(upr >= upl) update(1, 1, n, upl, upr, 1, 2);
        if(r + 1 >= R) break;
    }
    it = st[x].upper_bound(pa(L, L)); it--;
    pa e = *it;
    if(e.second >= mergeL) mergeL = e.first;
    it = st[x].upper_bound(pa(R, R)); it--;
    e = *it;
    if(e.second >= mergeR) mergeR = e.second;
    vector<pa> vec;
    it = st[x].lower_bound(pa(mergeL, mergeL));
    for(; it != st[x].end(); it++) {
        e = *it;
        if(e.first >= mergeL && e.second <= mergeR) vec.push_back(e);
        else break;
    }
    for(int i = 0; i < vec.size(); i++) st[x].erase(vec[i]);
    st[x].insert(pa(mergeL, mergeR));
}

int main() {
    ios::sync_with_stdio(0);
    cin >> n >> m;
    mod = 998244353; ll inv = (mod + 1) / 2;
    for(int i = 0; i <= n; i++) {
        st[i].insert(pa(0, 0));
        st[i].insert(pa(n + 1, n + 1));
    }
    build(1, 1, n);
    while(m--) {
        int op, l, r, x;
        cin >> op >> l >> r;
        if(op == 1) {
            cin >> x;
            update(1, 1, n, l, r, 2, 1);
            merge(x, l, r);
        } else {
            ll ans = query(1, 1, n, l, r);
            cout << ans << endl;
        }
    }
    return 0;
}