[2021牛客多校7 B] xay loves monotonicity（线段树）

题意

有长度为 $n$ 的序列 $a$ 和 $01$序列 $b$，有 $m$ 次操作，每次操作给出 $l,r$。每个操作为以下三种之一：

1. 将 $a_l$ 改成 $r$
2. 将 $b_l,b_{l+1}...,b_r$ 翻转 (即$b_i=(b_i+1)mod2$)
3. 设 $[l,r]$ 中 $a$ 形成的单调栈下标序列为 $p$，满足 $a_{p_k}\ge a_{p_{k-1}}$。求 $b_{p_k}\ne b_{p_{k+1}}$ 的数量。

其中， $1\le n\le 2\times 10^5,0\le a_i\le 10^9,0\le b_i\le 1$。

分析

令 $cal(l,r,x,p)$ 表示单调栈上一个的值为 $x$，位置为 $p$，经过 $l,r$ 之后多出来的答案。我们用线段树维护区间最大值 $mx$ 和区间最大值最靠右的位置 $mp$。
如果 $m{x}_{lson}<x$，那么 $cal(l,r,x,p)=cal(m+1,r,x,p)$。
如果 $m{x}_{lson}\ge x$，那么 $cal(l,r,x,p)=cal(l,m,x,p)+cal(m+1,r,m{x}_{lson},m{p}_{lson})$。
后面那部分是可以预处理的。这样子我们的询问每次只会往一边走，复杂度是 $O(logn)$ 的。
令 $CA{L}_{root}=cal(m+1,r,m{x}_{lson},m{p}_{lson})$。每次 $pushup$ 的时候用 $O(logn)$ 求 $CAL$ 即可。
注意 $b$ 数组的标记的处理就可以了。
总复杂度是 $O(nlo{g}^{2}n)$ 的。

代码如下

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

void debug_out(){
    cerr << endl;
}
template<typename Head, typename... Tail>
void debug_out(Head H, Tail... T){
    cerr << ' ' << H;
    debug_out(T...);
}
#ifdef local
#define debug(...) cerr << "[" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__)
#else
#define debug(...) 55
#endif

#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
#define lc rt << 1
#define rc rt << 1 | 1
const int N = 2e5 + 5;
int a[N], b[N];
int mx[N * 4], mp[N * 4], cal[N * 4], tag[N * 4], n;
void pushdown(int rt){
    if(tag[rt]){
        tag[lc] ^= 1;
        tag[rc] ^= 1;
        tag[rt] = 0;
    }
}
int qryb(int l, int r, int rt, int p){
    if(l == r) return (tag[rt] ^ b[l]);
    pushdown(rt);
    int m = l + r >> 1;
    if(p <= m) return qryb(lson, p);
    return qryb(rson, p);
}
int solve(int l, int r, int rt, int x, int p){
    if(l == r){
        if(mx[rt] >= x && (tag[rt] ^ b[l]) != qryb(1, n, 1, p)) return 1;
        return 0;
    }
    pushdown(rt);
    int m = l + r >> 1;
    if(x > mx[lc]) return solve(rson, x, p);
    return solve(lson, x, p) + cal[rt];
}
void pushup(int l, int r, int rt){
    int m = l + r >> 1;
    mx[rt] = max(mx[lc], mx[rc]);
    if(mx[rt] == mx[lc]) mp[rt] = mp[lc];
    if(mx[rt] == mx[rc]) mp[rt] = mp[rc];
    cal[rt] = solve(rson, mx[lc], mp[lc]);
}
void build(int l, int r, int rt){
    if(l == r){
        mx[rt] = a[l], mp[rt] = l;
        return;
    }
    int m = l + r >> 1;
    build(lson);
    build(rson);
    pushup(l, r, rt);
}
void upd1(int l, int r, int rt, int p, int c){
    if(l == r){
        mx[rt] = c;
        return;
    }
    pushdown(rt);
    int m = l + r >> 1;
    if(p <= m) upd1(lson, p, c);
    else upd1(rson, p, c);
    pushup(l, r, rt);
}
void upd2(int l, int r, int rt, int a, int b){
    if(l >= a && r <= b){
        tag[rt] ^= 1;
        return;
    }
    pushdown(rt);
    int m = l + r >> 1;
    if(a <= m) upd2(lson, a, b);
    if(b > m) upd2(rson, a, b);
    pushup(l, r, rt);
}
tuple<int, int, int> query(int l, int r, int rt, int x, int p, int a, int b){
    if(l >= a && r <= b){
        if(mx[rt] < x) return make_tuple(0, x, p);
        return make_tuple(solve(l, r, rt, x, p), mx[rt], mp[rt]);
    }
    pushdown(rt);
    int m = l + r >> 1;
    if(b <= m) return query(lson, x, p, a, b);
    if(a > m) return query(rson, x, p, a, b);
    auto p1 = query(lson, x, p, a, m);
    int a1, a2;
    tie(a1, x, p) = p1;
    auto p2 = query(rson, x, p, m + 1, b);
    tie(a2, x, p) = p2;
    return make_tuple(a1 + a2, x, p);
}
int main() {
#ifdef local
    freopen("../in.txt", "r", stdin);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    cin >> n;
    for(int i = 1; i <= n; i++) cin >> a[i];
    for(int i = 1; i <= n; i++) cin >> b[i];
    build(1, n, 1);
    int q;
    cin >> q;
    for(int i = 1, o, l, r; i <= q; i++){
        cin >> o >> l >> r;
        if(o == 1){
            a[l] = r;
            upd1(1, n, 1, l, r);
        }
        else if(o == 2){
            upd2(1, n, 1, l, r);
        }
        else{
            auto t = query(1, n, 1, a[l], l, l, r);
            int w, x, y;
            tie(w, x, y) = t;
            cout << w << '\n';
        }
    }
    return 0;
}