HDU - 5306 Gorgeous Sequence 线段树 + 均摊分析

Code: 

#include<algorithm> 
#include<cstdio>
#include<cstring>         
#define ll long long 
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 2000000 
#define lson (now<<1) 
#define rson ((now<<1)|1)
using namespace std;       
int n,q;  
ll arr[maxn],mx[maxn<<2],se[maxn<<2],cnt[maxn<<2],sum[maxn<<2];      
void pushup(int l,int r,int now) {   
    int mid=(l+r)>>1; 
    int ls=lson,rs=(r>mid)?rson:0;       
    sum[now]=sum[ls]+sum[rs];   
    if(mx[ls]>mx[rs]) mx[now]=mx[ls],cnt[now]=cnt[ls],se[now]=max(se[ls],mx[rs]);         
    if(mx[rs]>mx[ls]) mx[now]=mx[rs],cnt[now]=cnt[rs],se[now]=max(se[rs],mx[ls]);    
    if(mx[ls]==mx[rs]) mx[now]=mx[ls],cnt[now]=cnt[ls]+cnt[rs],se[now]=max(se[ls],se[rs]);         
} 
void mark(int now,ll v) {
    if(v<mx[now]) sum[now]-=(mx[now]-v)*cnt[now], mx[now]=v; 
}
void pushdown(int l,int r,int now) {      
    int mid=(l+r)>>1;  
    mark(lson,mx[now]); 
    if(r>mid) mark(rson,mx[now]);     
}
void build(int l,int r,int now) {
    if(l==r) {
        mx[now]=arr[l],se[now]=-1,sum[now]=arr[l],cnt[now]=1;       
        return;               
    } 
    int mid=(l+r)>>1;    
    build(l,mid,lson); 
    if(r>mid) build(mid+1,r,rson);   
    pushup(l,r,now);    
}           
void change(int l,int r,int now,int L,int R,ll v) { 
    if(mx[now]<=v) return; 
    if(l>=L&&r<=R&&se[now]<v) {
        mark(now,v);    
        return;  
    }  
    pushdown(l,r,now);    
    int mid=(l+r)>>1;    
    if(L<=mid) change(l,mid,lson,L,R,v);  
    if(R>mid) change(mid+1,r,rson,L,R,v);     
    pushup(l,r,now);     
} 
ll qmax(int l,int r,int now,int L,int R) {
    if(l>=L&&r<=R) return mx[now];   
    pushdown(l,r,now); 
    int mid=(l+r)>>1; 
    ll re=-1;      
    if(L<=mid) re=max(re,qmax(l,mid,lson,L,R)); 
    if(R>mid) re=max(re,qmax(mid+1,r,rson,L,R));    
    return re;      
} 
ll qsum(int l,int r,int now,int L,int R) {
    if(l>=L&&r<=R) return sum[now];  
    pushdown(l,r,now);  
    int mid=(l+r)>>1;   
    ll re=0;  
    if(L<=mid) re+=qsum(l,mid,lson,L,R);   
    if(R>mid) re+=qsum(mid+1,r,rson,L,R);    
    return re;     
}
void solve() {
    scanf("%d%d",&n,&q); 
    for(int i=1;i<=n;++i) scanf("%lld",&arr[i]);  
    build(1,n,1);    
    while(q--) {
        int opt,x,y; 
        ll t; 
        scanf("%d%d%d",&opt,&x,&y); 
        if(opt==0) scanf("%lld",&t),change(1,n,1,x,y,t);           
        if(opt==1) printf("%lld\n",qmax(1,n,1,x,y)); 
        if(opt==2) printf("%lld\n",qsum(1,n,1,x,y));      
    }
}
int main() {
    // setIO("input");  
    int T; 
    scanf("%d",&T);        
    mx[0]=se[0]=-1;                   
    while(T--) solve();   
    return 0; 
}