Codevs_P1080 线段树练习(线段树)

来源:互联网 发布:facebook总是网络错误 编辑:程序博客网 时间:2024/06/03 17:09

题目描述 Description
一行N个方格,开始每个格子里都有一个整数。现在动态地提出一些问题和修改:提问的形式是求某一个特定的子区间[a,b]中所有元素的和;修改的规则是指定某一个格子x,加上或者减去一个特定的值A。现在要求你能对每个提问作出正确的回答。1≤N<100000,,提问和修改的总数m<10000条。

输入描述 Input Description
输入文件第一行为一个整数N,接下来是n行n个整数,表示格子中原来的整数。接下一个正整数m,再接下来有m行,表示m个询问,第一个整数表示询问代号,询问代号1表示增加,后面的两个数x和A表示给位置X上的数值增加A,询问代号2表示区间求和,后面两个整数表示a和b,表示要求[a,b]之间的区间和。

输出描述 Output Description
共m行,每个整数

样例输入 Sample Input
6
4
5
6
2
1
3
4
1 3 5
2 1 4
1 1 9
2 2 6

样例输出 Sample Output
22
22

数据范围及提示 Data Size & Hint
1≤N≤100000, m≤10000

#include<cstdio>#include<iostream>using namespace std;#define N 100005int n,m,k;int a[N];typedef struct node;typedef node * tree;struct node{    int l,r,sum;    tree lc,rc;};void build(tree p,int l,int r){    p->l=l,p->r=r,p->sum=0;    if(l+1<r){        p->lc=new node;        p->rc=new node;        build(p->lc,l,((l+r)>>1));        build(p->rc,((l+r)>>1),r);        p->sum=p->lc->sum+p->rc->sum;    }    else{        p->lc=NULL,p->rc=NULL;p->sum=a[l];    }    return;}void change(tree p,int o,int data){    if(p->l+1==p->r) p->sum+=data;    else{        if(o<(p->l+p->r)/2) change(p->lc,o,data);        if(o>=(p->l+p->r)/2) change(p->rc,o,data);        p->sum=p->lc->sum+p->rc->sum;    }    return;}int sum(tree p,int l,int r){    if(p->l>=l&&p->r<=r) return p->sum;    int res=0;    if(l<(p->l+p->r)>>1) res+=sum(p->lc,l,r);    if(r>(p->l+p->r)>>1) res+=sum(p->rc,l,r);    return res;}int main(){    scanf("%d",&n);    for(int i=1;i<=n;i++) scanf("%d",&a[i]);    tree root=new node;    build(root,1,n+1);    scanf("%d",&m);int x,y;    for(int i=1;i<=m;i++){        scanf("%d",&k);        if(k==1){            scanf("%d%d",&x,&y);            change(root,x,y);        }        else{            scanf("%d%d",&x,&y);            printf("%d\n",sum(root,x,y+1));        }    }}
0 0
原创粉丝点击