bzoj4695 最假女选手(势能线段树/吉司机线段树)题解

题意:

已知(n)个数字,进行以下操作:

  • (1.)给一个区间([L,R]) 加上一个数(x)
  • (2.)把一个区间([L,R]) 里小于(x) 的数变成(x)
  • (3.)把一个区间([L,R]) 里大于(x) 的数变成(x)
  • (4.)求区间([L,R])的和
  • (5.)求区间([L,R])的最大值
  • (6.)求区间([L,R]) 的最小值

思路:

吉司机线段树。
假如我们要进行把一个区间([L,R]) 里小于(x) 的数变成(x)。那么我们可以维护一个最小值(Min)和次小值(sMin)和最小值数量(Minlen)。那么,当(Mingeq x)时,显然这个区间不需要操作;当(Min<x)(sMin>x)时,这时只要更新(Min=x);当(sMinleq x)时,继续往下(dfs)。操作(3)也是同理。
因为这里是没设重置的标记的,所以在(pushdown)时如果子节点和父节点产生冲突,那么以父节点为准。
复杂度(O(mlog^2n))(m)为操作数。
有点卡常。

代码:

#include<map>
#include<set>
#include<queue>
#include<cmath>
#include<stack>
#include<ctime>
#include<vector>
#include<cstdio>
#include<string>
#include<cstring>
#include<sstream>
#include<iostream>
#include<algorithm>
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
const int maxn = 5e5 + 5;
const int MAXM = 3e6;
const ll MOD = 998244353;
const ull seed = 131;
const int INF = 1 << 30;

#define lson (rt << 1)
#define rson (rt << 1 | 1)
int a[maxn];
int Max[maxn << 2], Min[maxn << 2], sMax[maxn << 2], sMin[maxn << 2];
int Maxlen[maxn << 2], Minlen[maxn << 2];
ll sum[maxn << 2];
int lazy[maxn << 2];
inline void pushup(int rt){ //这边建议加上inline
    sum[rt] = sum[lson] + sum[rson];

    if(Max[lson] < Max[rson]){
        Max[rt] = Max[rson];
        Maxlen[rt] = Maxlen[rson];
        sMax[rt] = max(sMax[rson], Max[lson]);
    }
    if(Max[lson] > Max[rson]){
        Max[rt] = Max[lson];
        Maxlen[rt] = Maxlen[lson];
        sMax[rt] = max(sMax[lson], Max[rson]);
    }
    if(Max[lson] == Max[rson]){
        Max[rt] = Max[lson];
        Maxlen[rt] = Maxlen[lson] + Maxlen[rson];
        sMax[rt] = max(sMax[lson], sMax[rson]);
    }

    if(Min[lson] > Min[rson]){
        Min[rt] = Min[rson];
        Minlen[rt] = Minlen[rson];
        sMin[rt] = min(sMin[rson], Min[lson]);
    }
    if(Min[lson] < Min[rson]){
        Min[rt] = Min[lson];
        Minlen[rt] = Minlen[lson];
        sMin[rt] = min(sMin[lson], Min[rson]);
    }
    if(Min[lson] == Min[rson]){
        Min[rt] = Min[lson];
        Minlen[rt] = Minlen[lson] + Minlen[rson];
        sMin[rt] = min(sMin[lson], sMin[rson]);
    }
}
inline void pushdown(int rt, int l, int r){
    int m = (l + r) >> 1;
    if(lazy[rt]){
        sum[lson] += 1LL * (m - l + 1) * lazy[rt];
        sum[rson] += 1LL * (r - m) * lazy[rt];
        Max[lson] += lazy[rt];
        Max[rson] += lazy[rt];
        sMax[lson] += lazy[rt];
        sMax[rson] += lazy[rt];
        Min[lson] += lazy[rt];
        Min[rson] += lazy[rt];
        sMin[lson] += lazy[rt];
        sMin[rson] += lazy[rt];
        lazy[lson] += lazy[rt];
        lazy[rson] += lazy[rt];
        lazy[rt] = 0;
    }
    if(Max[lson] > Max[rt]){	//要和父节点保持一致
        if(sMax[lson] == Max[lson]) sMax[lson] = Max[rt];
        if(Min[lson] == Max[lson]) Min[lson] = Max[rt];
        if(sMin[lson] == Max[lson]) sMin[lson] = Max[rt];
        sum[lson] += 1LL * (Max[rt] - Max[lson]) * Maxlen[lson];
        Max[lson] = Max[rt];
    }
    if(Max[rson] > Max[rt]){
        if(sMax[rson] == Max[rson]) sMax[rson] = Max[rt];
        if(Min[rson] == Max[rson]) Min[rson] = Max[rt];
        if(sMin[rson] == Max[rson]) sMin[rson] = Max[rt];
        sum[rson] += 1LL * (Max[rt] - Max[rson]) * Maxlen[rson];
        Max[rson] = Max[rt];
    }
    if(Min[lson] < Min[rt]){
        if(sMin[lson] == Min[lson]) sMin[lson] = Min[rt];
        if(Max[lson] == Min[lson]) Max[lson] = Min[rt];
        if(sMax[lson] == Min[lson]) sMax[lson] = Min[rt];
        sum[lson] += 1LL * (Min[rt] - Min[lson]) * Minlen[lson];
        Min[lson] = Min[rt];
    }
    if(Min[rson] < Min[rt]){
        if(sMin[rson] == Min[rson]) sMin[rson] = Min[rt];
        if(Max[rson] == Min[rson]) Max[rson] = Min[rt];
        if(sMax[rson] == Min[rson]) sMax[rson] = Min[rt];
        sum[rson] += 1LL * (Min[rt] - Min[rson]) * Minlen[rson];
        Min[rson] = Min[rt];
    }
}
void build(int l, int r, int rt){
    lazy[rt] = 0;
    if(l == r){
        sum[rt] = Max[rt] = Min[rt] = a[l];
        sMax[rt] = -INF;
        sMin[rt] = INF;
        Maxlen[rt] = Minlen[rt] = 1;
        return;
    }
    int m = (l + r) >> 1;
    build(l, m, lson);
    build(m + 1, r, rson);
    pushup(rt);
}
void add(int L, int R, int l, int r, int v, int rt){
    if(L <= l && R >= r){
        sum[rt] += 1LL * v * (r - l + 1);
        Max[rt] += v;
        Min[rt] += v;
        sMax[rt] += v;
        sMin[rt] += v;
        lazy[rt] += v;
        return;
    }
    pushdown(rt, l, r);
    int m = (l + r) >> 1;
    if(L <= m)
        add(L, R, l, m, v, lson);
    if(R > m)
        add(L, R, m + 1, r, v, rson);
    pushup(rt);
}
void Less(int L, int R, int l, int r, int v, int rt){
    if(Min[rt] >= v) return;
    if(L <= l && R >= r && sMin[rt] > v){//>保证Minlen不变
        if(Max[rt] == Min[rt]) Max[rt] = v;
        if(sMax[rt] == Min[rt]) sMax[rt] = v;
        sum[rt] += 1LL * (v - Min[rt]) * Minlen[rt] ;
        Min[rt] = v;
        return;
    }
    pushdown(rt, l, r);
    int m = (l + r) >> 1;
    if(L <= m)
        Less(L, R, l, m, v, lson);
    if(R > m)
        Less(L, R, m + 1, r, v, rson);
    pushup(rt);
}
void More(int L, int R, int l, int r, int v, int rt){
    if(Max[rt] <= v) return;
    if(L <= l && R >= r && sMax[rt] < v){   //<保证Maxlen不变
        if(Min[rt] == Max[rt]) Min[rt] = v;
        if(sMin[rt] == Max[rt]) sMin[rt] = v;
        sum[rt] += 1LL * (v - Max[rt]) * Maxlen[rt];
        Max[rt] = v;
        return;
    }
    pushdown(rt, l, r);
    int m = (l + r) >> 1;
    if(L <= m)
        More(L, R, l, m, v, lson);
    if(R > m)
        More(L, R, m + 1, r, v, rson);
    pushup(rt);
}
ll querySum(int L, int R, int l, int r, int rt){
    if(L <= l && R >= r){
        return sum[rt];
    }
    pushdown(rt, l, r);
    int m = (l + r) >> 1;
    ll ret = 0;
    if(L <= m)
        ret += querySum(L, R, l, m, lson);
    if(R > m)
        ret += querySum(L, R, m + 1, r, rson);
    return ret;
}
int queryMax(int L, int R, int l, int r, int rt){
    if(L <= l && R >= r){
        return Max[rt];
    }
    pushdown(rt, l, r);
    int m = (l + r) >> 1;
    int MAX = -INF;
    if(L <= m)
        MAX = max(MAX, queryMax(L, R, l, m, lson));
    if(R > m)
        MAX = max(MAX, queryMax(L, R, m + 1, r, rson));
    return MAX;
}
int queryMin(int L, int R, int l, int r, int rt){
    if(L <= l && R >= r){
        return Min[rt];
    }
    pushdown(rt, l, r);
    int m = (l + r) >> 1;
    int MIN = INF;
    if(L <= m)
        MIN = min(MIN, queryMin(L, R, l, m, lson));
    if(R > m)
        MIN = min(MIN, queryMin(L, R, m + 1, r, rson));
    return MIN;
}
inline bool read(int &num){
        char in;
        bool IsN=false;
        in = getchar();
        if(in == EOF) return false;
        while(in != '-' && (in < '0' || in > '9')) in = getchar();
        if(in == '-'){ IsN = true; num = 0;}
        else num = in - '0';
        while(in = getchar(),in >= '0' && in <= '9'){
                num *= 10, num += in-'0';
        }
        if(IsN) num = -num;
        return true;
}

int main(){
    int n;
    read(n);
    for(int i = 1; i <= n; i++) read(a[i]);
    build(1, n, 1);
    int m;
    read(m);
    while(m--){
        int l, r, x, op;
        read(op), read(l), read(r);
        if(op <= 3) read(x);
        if(op == 1) add(l, r, 1, n, x, 1);
        else if(op == 2) Less(l, r, 1, n, x, 1);
        else if(op == 3) More(l, r, 1, n, x, 1);
        else if(op == 4) printf("%lld
", querySum(l, r, 1, n, 1));
        else if(op == 5) printf("%d
", queryMax(l, r, 1, n, 1));
        else printf("%d
", queryMin(l, r, 1, n, 1));
    }
    return 0;
}

原文地址:https://www.cnblogs.com/KirinSB/p/11506002.html