线段树模版(codevs1081线段树练习2)

题目描述 Description
给你N个数，有两种操作

1：给区间[a,b]的所有数都增加X

2：询问第i个数是什么？

输入描述 Input Description
第一行一个正整数n，接下来n行n个整数，再接下来一个正整数Q，表示操作的个数. 接下来Q行每行若干个整数。如果第一个数是1，后接3个正整数a,b,X，表示在区间[a,b]内每个数增加X,如果是2，后面跟1个整数i, 表示询问第i个位置的数是多少。

输出描述 Output Description
对于每个询问输出一行一个答案

样例输入 Sample Input
3

1

2

3

2

1 2 3 2

2 3

样例输出 Sample Output
5

数据范围及提示 Data Size & Hint
数据范围

1<=n<=100000

1<=q<=100000

#include<cstdio>
#include<algorithm>
using namespace std;
int tree[100000*4],a[100000],lazy[100000*4];
int build(int l,int r,int q){
    if(l==r) return tree[q]=a[l];
    int mid=(l+r)/2;
    return tree[q]=build(l,mid,q*2)+build(mid+1,r,q*2+1);
}
void putdown(int q,int l,int r){
    int mid=(l+r)/2;
    tree[q*2]+=lazy[q]*(mid-l+1);
    tree[q*2+1]+=lazy[q]*(r-mid);
    lazy[q*2]+=lazy[q];
    lazy[q*2+1]+=lazy[q];
    lazy[q]=0;
}
int query(int a,int b,int l,int r,int q){
    if(a<=l&&r<=b) return tree[q];
    if(b<l||r<a) return 0;
    putdown(q,l,r);
    int mid=(l+r)/2;
    return query(a,b,l,mid,q*2)+query(a,b,mid+1,r,q*2+1);
}
void update(int a,int b,int l,int r,int q,int x){
    if(b<l||r<a) return ;
    if(a<=l&&r<=b) {
        tree[q]+=(r-l+1)*x;
        lazy[q]+=x;
    }
    else{
        int mid=(l+r)/2;
        putdown(q,l,r);
        update(a,b,l,mid,q*2,x);
        update(a,b,mid+1,r,q*2+1,x);
    }
}
int main(){
    freopen("segtree.in","r",stdin);
    freopen("segtree.out","w",stdout);
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    build(1,n,1);
    int q;
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        int s;
        scanf("%d",&s);
        if(s==1) {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            update(x,y,1,n,1,z);
        }
        else{
            int x,y;
            scanf("%d%d",&x,&y);
            printf("%d\n",query(x,y,1,n,1));
        }
    }
    return 0;
}