题目描述 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<iostream> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int maxn=100000+5; inline int read() { int x=0,f=1; char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0'; ch=getchar();} return x*f; } int n,m,len; int a[maxn]; struct node { int l,r,lc,rc,c; }tr[maxn<<1]; inline void bt(int x,int y) { len++; int now=len; tr[now].l=x;tr[now].r=y;tr[now].c=0; if(x==y) tr[now].c=a[x]; else { int mid=(x+y)>>1; tr[now].lc=len+1; bt(x,mid); tr[now].rc=len+1; bt(mid+1,y); tr[now].c=tr[tr[now].lc].c+tr[tr[now].rc].c; } } inline void update(int now,int x,int k) { int lc=tr[now].lc,rc=tr[now].rc; if(tr[now].l==tr[now].r) tr[now].c+=k; else { int mid=(tr[now].l+tr[now].r)>>1; if(x<=mid) update(lc,x,k); else if(x>=mid+1) update(rc,x,k); tr[now].c=tr[lc].c+tr[rc].c; } } inline int query(int now,int x,int y) { int lc=tr[now].lc,rc=tr[now].rc; if(tr[now].l==x&&tr[now].r==y) return tr[now].c; else { int mid=(tr[now].l+tr[now].r)>>1; if(y<=mid) return query(lc,x,y); else if(x>=mid+1) return query(rc,x,y); else return query(lc,x,mid)+query(rc,mid+1,y); } } int main() { n=read(); for(int i=1;i<=n;i++) a[i]=read(); bt(1,n); m=read(); for(int i=1;i<=m;i++) { int p,x,y; p=read();x=read();y=read(); if(p==1) update(1,x,y); else printf("%d ",query(1,x,y)); } return 0; }