CodeVs1080

传送门：线段树练习

单点修改 + 区间查询

题目描述  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 。

ZKW 线段树 代码

#include<iostream> #include<cstdio> using namespace std; int a[100000*4+100]; int n,m,M; void build() { for(M=1;M<=n+2;M<<=1); for(int i=M+1;i<=M+n;i++) { scanf("%d",&a[i]); } for(int i=M-1;i>=1;i--) { a[i] = a[i<<1] + a[i<<1|1]; } } void update(int x,int k) { x+=M; a[x] += k; for(x>>=1;x;x>>=1) a[x] = a[x<<1] + a[x<<1|1]; } int query(int l,int r) { int ans = 0; l+=M-1; r+=M+1; while(l^r^1) { if(~l&1) ans += a[l^1]; if(r&1) ans += a[r^1]; l>>=1; r>>=1; } return ans; } int main() { scanf("%d",&n); build(); scanf("%d",&m); while(m--) { int op,a,b; scanf("%d%d%d",&op,&a,&b); if(op == 1) { update(a,b); } else { printf("%d\n",query(a,b)); } } return 0; }

递归版线段树代码

#include<cstdio> #include<iostream> #define N 400100 int n,m,x,y,p,ans; struct No{ int l; int r; int v; }tree[N]; void build(int l,int r,int k) { tree[k].l = l; tree[k].r = r; if(tree[k].l == tree[k].r) { scanf("%d",&tree[k].v); return ; } int m = (tree[k].l + tree[k].r )>>1; build(l,m,k<<1); build(m+1,r,k<<1|1); tree[k].v = tree[k<<1].v + tree[k<<1|1].v; } void add(int k) { if(tree[k].l == tree[k].r) { tree[k].v += y; return ; } int m = (tree[k].l + tree[k].r) >>1; if(x<=m) add(k<<1); else add(k<<1|1); tree[k].v = tree[k<<1].v + tree[k<<1|1].v; } void ask(int k) { if(tree[k].l >=x&&tree[k].r <=y) { ans += tree[k].v; return ; } int m =(tree[k].l + tree[k].r )>>1; if(x<=m) ask(k<<1); if(y>m) ask(k<<1|1); } int main() { scanf("%d",&n); build(1,n,1); scanf("%d",&m); while(m--) { scanf("%d%d%d",&p,&x,&y); ans =0; if(p==1) add(1); else { ask(1); printf("%d\n",ans); } } return 0; }

当然。。。这个题还是树状数组的模板。。。

#include<iostream> #include<cstdio> using namespace std; int n,m; int a[1000000]; int lowbit(int x) { return x & -x; } void update(int p,int x) { while(p<=n) { a[p] += x; p += lowbit(p); } } int query(int p) { int sum = 0; while(p) { sum += a[p]; p -= lowbit(p); } return sum; } int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { int tt; scanf("%d",&tt); update(i,tt); } scanf("%d",&m); while(m--) { int op,a,b; scanf("%d%d%d",&op,&a,&b); if(op == 1) { update(a,b); } else { printf("%d\n",query(b) - query(a-1)); } } }