题目描述 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 。
思路:
虽然树状数组能AC但是还是想要用线段树撸出来于是就有了这篇。我们为线段树每个节点增加一个Count域表示所对应区间内元素之和。每次修改一个格子,需要修改从叶结点到根结点路径上所有结点的值。开始将m个元素一个个加入。注意这里区间[l,r]表示l到r-1个格子,并不是l到r。
源代码/pas:
type
tree=record
l,r,c,g:longint;
end;
var
t:array[1..400000]of tree;
d,p:longint;
procedure maketree(f:Longint);
var
mid:longint;
begin
if t[f].r-t[f].l>1 then
begin
mid:=(t[f].l+t[f].r)div 2;
t[f*2].l:=t[f].l;
t[f*2].r:=mid;
t[f*2+1].l:=mid;
t[f*2+1].r:=t[f].r;
maketree(f*2);
maketree(f*2+1);
end
else
if t[f].r-t[f].l=1 then
begin
inc(p);
t[f].g:=p;
end;
end;
function count(f,x,y:longint):longint;
var
mid:longint;
begin
if (t[f].l=x)and(t[f].r=y) then
count:=t[f].c
else
begin
mid:=(t[f].l+t[f].r)div 2;
if y<=mid then
count:=count(f*2,x,y)
else
if x>=mid then
count:=count(f*2+1,x,y)
else
count:=count(f*2,x,mid)+count(f*2+1,mid,y);
end;
end;
procedure find(f,x:Longint);
begin
if t[f].r-t[f].l=1 then
begin
if t[f].g=x then d:=f;
exit;
end
else
begin
find(f*2,x);
find(f*2+1,x);
end;
end;
procedure add(x,y:longint);
begin
repeat
t[x].c:=t[x].c+y;
x:=x div 2;
until x=0;
end;
procedure main;
var
n,m,i,j,k,v:longint;
x,y,z:longint;
begin
readln(n);
t[1].l:=1;
t[1].r:=n+1;
maketree(1);
for i:=1 to n do
begin
readln(v);
find(1,i);
add(d,v);
end;
readln(m);
for i:=1 to m do
begin
readln(z,x,y);
case z of
2:writeln(count(1,x,y+1));
1:
begin
find(1,x);
add(d,y);
end;
end;
end;
end;
begin
main;
end.