[HDU5306]吉司机线段树

吉司机线段树

给每一个点定义一个势能函数为该节点下不同的值的个数

若该操作能够使当前节点的势能降低，那么暴力修改该区间内的所有元素，否则直接打标记

这题每个节点记录最大值mx，最大值个数tot，次大值se，答案sum；

若新来的值比>=mx，那么这个操作是没有意义的，直接return

若新来的值>se并且

若新来的值<=se，那么这次操作会使该节点势能函数降低，暴力递归修改区间内所有元素

tips：

这题卡常，要用读入挂才能过

#include 
#pragma GCC optimize (2)

#define mid ((l+r)>>1)
#define ls l,mid,t<<1
#define rs mid+1,r,t<<1|1

#define N 1000050

typedef long long LL;
struct Node{int mx,tot,se;LL sum;}tr[4*N];
int a[N],n,m;
int ag[4*N];

char *ch, *ch1, buf[40*1024000+5], buf1[40*1024000+5];  
  
void read(int &x)  
{  
    for (++ch; *ch <= 32; ++ch);  
    for (x = 0; '0' <= *ch; ch++)    x = x * 10 + *ch - '0';  
} 

inline int max(const int &a,const int &b){return a tr[t].mx) { ag[t] = -1; return ; }
    if (ag[t] <= tr[t].se) {
        if (ag[t] != -1 && l != r) push_down(l,r,t);
        add( tr[t] , tr[t<<1] , tr[t<<1|1] );
        return ; 
    }
    tr[t].sum -= (LL)(tr[t].mx - ag[t]) * tr[t].tot;
    tr[t].mx = ag[t];
    
}

inline void push_down(int l,int r,int t) {
    ag[t<<1] = (ag[t<<1] != -1) ? min(ag[t<<1],ag[t]) : ag[t];
    ag[t<<1|1] = (ag[t<<1|1] != -1) ? min(ag[t<<1|1],ag[t]) : ag[t];
    ag[t] = -1;
    color(ls); color(rs);
}

int ll,rr,v;
void update(int l,int r,int t) {
    if (ll <= l && r <= rr) {
        ag[t] = (ag[t] != -1) ? min(ag[t],v) : v;
        color(l,r,t);
        return ;
    } 
    if (ag[t] != -1) push_down(l,r,t);
    if (ll <= mid) update(ls);
    if (rr > mid) update(rs);
    add( tr[t] , tr[t<<1] , tr[t<<1|1] );
}
LL ans_sum;int ans_max;
void query(int l,int r,int t) {
    if (ll <= l && r <= rr) { 
        ans_sum += tr[t].sum;
        ans_max = max(tr[t].mx , ans_max);
        return ; 
    }
    if (ag[t] != -1) push_down(l,r,t);
    if (ll <= mid) query(ls);
    if (mid < rr) query(rs);
    add( tr[t] , tr[t<<1] , tr[t<<1|1] );
}

void solve() {
//    memset(tr,0,sizeof(tr));
//    memset(ag,0,sizeof(ag));
    read(n); read(m);
    for (int i=1;i<=n;i++) read(a[i]);
    build(1,n,1);
    for (int _=1,cmd;_<=m;_++) {
        read(cmd);
        if (cmd == 0) {
            read(ll); read(rr); read(v);
            update(1,n,1);
        } else {
            ans_sum = ans_max = 0;
            read(ll); read(rr);
            query(1,n,1);
            if (cmd == 1) 
                print(ans_max);
            else
                print(ans_sum);
            putchar('\n');
        }
    }
}
int main() {
    ch = buf - 1;  
    ch1 = buf1 - 1;  
    fread(buf, 1, 1000 * 35 * 1024, stdin);  
    int T;
    for (read(T);T--;) solve();
    return 0;
} 

给每一个点定义一个势能函数