codevs4927 线段树练习5

题目描述 Description

有n个数和5种操作

add a b c：把区间[a,b]内的所有数都增加c

set a b c：把区间[a,b]内的所有数都设为c

sum a b：查询区间[a,b]的区间和

max a b：查询区间[a,b]的最大值

min a b：查询区间[a,b]的最小值

输入描述 Input Description

第一行两个整数n，m，第二行n个整数表示这n个数的初始值

接下来m行操作，同题目描述

输出描述 Output Description

对于所有的sum、max、min询问，一行输出一个答案

样例输入 Sample Input

10 6

3 9 2 8 1 7 5 0 4 6

add 4 9 4

set 2 6 2

add 3 8 2

sum 2 10

max 1 7

min 3 6

样例输出 Sample Output

49

11

4

数据范围及提示 Data Size & Hint

10%:1<n,m<=10

30%:1<n,m<=10000

100%:1<n,m<=100000

保证中间结果在long long(C/C++)、int64(pascal)范围内

 

正解：线段树

解题报告：

　　这就是传说中的线段树综合题，所有标记都有。

　　注意一点，就是我的执行顺序是先add标记下传，然后才是set标记下传，但下传过程中set可以覆盖add。开始我一直WA，后来才发现了一个错误，就是我先执行的add，那么，如果当前结点上此时既有set又有add，那么只能说明add一定是在set之后执行的，不然执行set的时候add早已经下传，所以我可以考虑直接把add的值直接加在set上去，就可以避免有一部分add没有发挥作用。害得我调试了半个小时。

 

//It is made by jump~
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <set>
using namespace std;
typedef long long LL;
#define RG register
const int MAXN = 100011;
int n,m,ql,qr;
LL val,ans,inf;
char ch[10];
struct node{
    int l,r;
    LL minl,maxl,sum,add,set;
    bool flag;
}a[MAXN*3];

inline LL getint(){RG LL w=0,q=0; char c=getchar();while((c<'0' || c>'9') && c!='-') c=getchar(); if(c=='-') q=1,c=getchar();while (c>='0' && c<='9') w=w*10+c-'0', c=getchar(); return q ? -w : w;}

inline void update(RG int root){
    RG int lc=root*2,rc=lc+1;  a[root].maxl=max(a[lc].maxl,a[rc].maxl);
    a[root].sum=a[lc].sum+a[rc].sum;  a[root].minl=min(a[lc].minl,a[rc].minl);
}

inline void build(RG int root,RG int l,RG int r){
    a[root].l=l; a[root].r=r;
    if(l==r) {    a[root].minl=a[root].maxl=a[root].sum=getint();    return ;  }
    RG int mid=(l+r)/2; RG int lc=root*2,rc=lc+1;
    build(lc,l,mid); build(rc,mid+1,r); update(root);
}

inline void pushdown(int root,int l,int r){
    if(!a[root].flag) return ; if(l==r) { a[root].flag=false; a[root].set=0; return ; }
    int lc=root*2,rc=lc+1; int mid=(l+r)/2;
    a[lc].set=a[rc].set=a[root].set; a[root].add=a[lc].add=a[rc].add=0;
    a[lc].maxl=a[rc].maxl=a[lc].minl=a[rc].minl=a[root].set;
    a[lc].sum=(LL)(mid-l+1)*a[root].set; a[rc].sum=(LL)(r-mid)*a[root].set;  a[root].sum=a[lc].sum+a[rc].sum;
    a[lc].flag=a[rc].flag=true;
    a[root].set=0; a[root].flag=false;
}

inline void pushdown_add(RG int root,RG int l,RG int r){
    if(l==r) { a[root].add=0; return ; } if(a[root].add==0) return ;
 
    if(a[root].flag){ a[root].set+=a[root].add; a[root].add=0; return ; }//至关重要，因为add打在set的后面，这就意味着相当于我们可以把set设的值变大一点

    RG int mid=(l+r)/2; RG int lc=root*2,rc=lc+1; a[lc].add+=a[root].add; a[rc].add+=a[root].add;    
    a[lc].minl+=a[root].add; a[lc].maxl+=a[root].add; a[rc].minl+=a[root].add; a[rc].maxl+=a[root].add;
    a[lc].sum+=a[root].add*(LL)(mid-l+1); a[rc].sum+=a[root].add*(LL)(r-mid);
    a[root].add=0; a[root].sum=a[lc].sum+a[rc].sum;
}

inline void add(RG int root,RG int l,RG int r){
    pushdown_add(root,l,r); pushdown(root,l,r); 
    if(ql<=l && r<=qr) { a[root].add+=val; a[root].maxl+=val; a[root].minl+=val; a[root].sum+=val*(LL)(r-l+1); return ; }
    RG int mid=(l+r)/2; RG int lc=root*2,rc=lc+1; if(ql<=mid) add(lc,l,mid); if(qr>mid) add(rc,mid+1,r);
    update(root);
}

inline void update_set(RG int root,RG int l,RG int r){    
    if(ql<=l && r<=qr) { a[root].flag=true; a[root].minl=a[root].maxl=a[root].set=val; a[root].add=0; a[root].sum=(LL)(r-l+1)*val; return ; } 
    pushdown_add(root,l,r); pushdown(root,l,r);
    RG int mid=(l+r)/2;RG int lc=root*2,rc=lc+1;
    if(ql<=mid) update_set(lc,l,mid); if(qr>mid) update_set(rc,mid+1,r); update(root);
}

inline void query_sum(RG int root,RG int l,RG int r){
    pushdown_add(root,l,r);  pushdown(root,l,r);
    if(ql<=l && r<=qr){ ans+=a[root].sum; return ; }
    RG int mid=(l+r)/2; RG int lc=root*2,rc=lc+1; 
    if(ql<=mid) query_sum(lc,l,mid); if(qr>mid) query_sum(rc,mid+1,r); update(root);
}

inline void query_max(RG int root,RG int l,RG int r){
    pushdown_add(root,l,r); pushdown(root,l,r);
    if(ql<=l && r<=qr){ ans=max(a[root].maxl,ans); return ; }
    RG int mid=(l+r)/2; RG int lc=root*2,rc=lc+1; 
    if(ql<=mid) query_max(lc,l,mid); if(qr>mid) query_max(rc,mid+1,r); update(root);
}

inline void query_min(RG int root,RG int l,RG int r){
    pushdown_add(root,l,r); pushdown(root,l,r);
    if(ql<=l && r<=qr){ ans=min(a[root].minl,ans); return ; }
    RG int mid=(l+r)/2; RG int lc=root*2,rc=lc+1; 
    if(ql<=mid) query_min(lc,l,mid); if(qr>mid) query_min(rc,mid+1,r); update(root);
}

inline void work(){
    n=getint(); m=getint(); build(1,1,n); inf=1; for(RG int i=1;i<=62;i++) inf*=2;
    while(m--) {
    scanf("%s",ch);
    if(ch[0]=='a') { ql=getint(); qr=getint(); val=getint(); add(1,1,n); }
    else if(ch[0]=='s' && ch[1]=='e') { ql=getint(); qr=getint(); val=getint(); update_set(1,1,n); }
    else if(ch[0]=='s' && ch[1]=='u') { ql=getint(); qr=getint(); ans=0; query_sum(1,1,n); printf("%lld
",ans); }
    else if(ch[0]=='m' && ch[1]=='a') { ql=getint(); qr=getint(); ans=-inf; query_max(1,1,n); printf("%lld
",ans); }
    else{ ql=getint(); qr=getint(); ans=inf; query_min(1,1,n); printf("%lld
",ans); }
    }
}

int main()
{
  work();
  return 0;
}