线段树练习_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 。

思路：

虽然树状数组能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.

转载于:https://www.cnblogs.com/olahiuj/p/5781322.html