洛谷OJ U552 守墓人 线段树模板题

题目描述 Description

在一个荒凉的墓地上
有一个令人尊敬的守墓人， 他看守的墓地从来
没有被盗过， 所以人们很放心的把自己的先人的墓
安顿在他那
守墓人能看好这片墓地是必然而不是偶然.....
因为....守墓人懂风水 0.0
他把墓地分为主要墓碑和次要墓碑， 主要墓碑
只能有 1 个， 守墓人把他记为 1 号， 而次要墓碑有
n-1 个，守墓人将之编号为 2,3...n，所以构成了一个有 n 个墓碑的墓地。
而每个墓碑有一个初始的风水值，这些风水值决定了墓地的风水的好坏，所以守墓人
需要经常来查询这些墓碑。
善于运用风水的守墓人，通过一次次逆天改命，使得自己拥有了无限寿命，没人知道
他活了多久。
这天，你幸运的拜访到了他，他要求你和他共同见证接下来几年他的战果，但不过他
每次统计风水值之和都需要你来帮他计算，算错了他会要你命 QAQ
风水也不是不可变，除非遭遇特殊情况，已知在接下来的 2147483647 年里，会有 n 次
灾难，守墓人会有几个操作：
1.将[l,r]这个区间所有的墓碑的风水值增加 k。
2.将主墓碑的风水值增加 k
3.将主墓碑的风水值减少 k
4.统计[l,r]这个区间所有的墓碑的风水值之和
5.求主墓碑的风水值
上面也说了，很多人会把先人的墓安居在这里，而且守墓人活了很多世纪→_→，墓碑
的数量会多的你不敢相信= =
守墓人和善的邀请你帮他完成这些操作，要不然哪天你的旅馆爆炸了，天上下刀子.....
为了活命，还是帮他吧

输入输出格式 Input/output

输入格式：
第一行，两个正整数 n，f 表示共有 n 块墓碑，并且在接下来的
2147483647 年里，会有 f 次世界末日
第二行，n 个正整数，表示第 i 块墓碑的风水值
接下来 f 行，每行都会有一个针对世界末日的解决方案，如题所述，标记同题

输出格式：

输出会有若干行，对 4 和 5 的提问做出回答

输入输出样例 Sample input/output

样例测试点#1

输入样例：

5 6
0 0 0 0 0
1 1 5 1
1 1 3 3
2 3
3 1
4 1
5

输出样例：

16
7

说明 description

20%的数据满足：1≤n≤100
50%的数据满足：1≤n≤6000
100%的数据满足：1≤n,f≤2*10^5

#include <iostream>
#include <cstdio>
using namespace std;

#define ls id<<1
#define rs (id<<1)+1
#define lson l,mid,ls
#define rson mid+1,r,rs

const int maxn=200001;

struct segment_tree{
    int l,r;
    long long tag,num;
    segment_tree(){tag=num=l=r=0;}
}t[maxn*4];

int n,f;
int num[maxn];

void push_up(int id){
    t[id].num=t[ls].num+t[rs].num;
}

void push_down(int id){
    if(t[id].tag){
        t[ls].tag+=t[id].tag;
        t[rs].tag+=t[id].tag;
        t[ls].num+=t[id].tag*(t[ls].r-t[ls].l+1);
        t[rs].num+=t[id].tag*(t[rs].r-t[rs].l+1);
        t[id].tag=0;
    }
}

void build(int l,int r,int id){
    t[id].l=l;t[id].r=r;
    if(l==r){
        t[id].num=num[l];
        t[id].tag=0;
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    push_up(id);
}

long long query(int L,int R,int id){
    if(L<=t[id].l && t[id].r<=R)return t[id].num;
    push_down(id);
    long long res=0;
    int mid=(t[id].l+t[id].r)>>1;
    if(mid<L)
       res=query(L,R,rs); 
    else
      if(mid>=R)
        res=query(L,R,ls);
      else 
        res=query(L,mid,ls)+query(mid+1,R,rs);
    return res;
}

void update(int L,int R,int id,int c){
    if(L<=t[id].l && t[id].r<=R){
        t[id].num+=(t[id].r-t[id].l+1)*c;
        t[id].tag+=c;
        return;
    }
    push_down(id);
    if(R>=t[rs].l)update(L,R,rs,c);
    if(L<=t[ls].r)update(L,R,ls,c);
    push_up(id);
}

int main(){
    scanf("%d%d",&n,&f);
    int i,x,l,r,k;
    for(i=1;i<=n;i++)
      scanf("%d",&num[i]);
    build(1,n,1);
    for(i=1;i<=f;i++){
        scanf("%d",&x);
        if(x==1){
            scanf("%d%d%d",&l,&r,&k);
            update(l,r,1,k);
        }
        if(x==2){
            scanf("%d",&k);
            update(1,1,1,k);
        }
        if(x==3){
            scanf("%d",&k);
            update(1,1,1,-k);
        }
        if(x==4){
            scanf("%d%d",&l,&r);
            printf("%lld\n",query(l,r,1));
        }
        if(x==5)printf("%lld\n",query(1,1,1));
    }
    return 0;
}