[CodeVS1081]线段树练习2(区间修改+单点询问)

线段树练习2

题目描述 Description

给你N个数，有两种操作

1：给区间[a,b]的所有数都增加X

2：询问第i个数是什么？

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，再接下来一个正整数Q，表示操作的个数. 接下来Q行每行若干个整数。如果第一个数是1，后接3个正整数a,b,X，表示在区间[a,b]内每个数增加X,如果是2，后面跟1个整数i, 表示询问第i个位置的数是多少。

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3

1

2

3

2

1 2 3 2

2 3

样例输出 Sample Output

5

数据范围及提示 Data Size & Hint

1<=n<=100000

1<=q<=100000

var
w:array[0..400005,1..4]of longint;
x:array[0..100000]of longint;
i,j,k:longint;
n,m:longint;
a,b,c,d:longint;
procedure build(a,l,r:longint);
var mid:longint;
begin
w[a,1]:=l; w[a,2]:=r;
if l=r then begin w[a,3]:=x[l]; exit; end;
mid:=(l+r)div 2;
build(a*2,l,mid);
build(a*2+1,mid+1,r);
end;

procedure update(a,l,r:longint);
var mid:longint;
begin
if (w[a,1]=l)and(w[a,2]=r)
then begin inc(w[a,3],d); exit; end;
mid:=(w[a,1]+w[a,2])div 2;
if r<=mid
then update(a*2,l,r) else
if l>mid
then update(a*2+1,l,r) else
begin
update(a*2,l,mid);
update(a*2+1,mid+1,r);
end;
end;

function query(a,b:longint):longint;
var mid:longint;
begin
if (w[a,1]<>w[a,2])and(w[a,3]<>0) then begin inc(w[a*2,3],w[a,3]); inc(w[a*2+1,3],w[a,3]); w[a,3]:=0; end;
if (w[a,1]=w[a,2])and(w[a,1]=b)
then exit(w[a,3]);
mid:=(w[a,1]+w[a,2])div 2;
if b<=mid
then exit(query(a*2,b))
else exit(query(a*2+1,b));
end;

begin
readln(n);
for i:=1 to n do
readln(x[i]);
build(1,1,n);
readln(m);
for i:=1 to m do
begin
read(a);
if a=1
then begin readln(b,c,d); update(1,b,c); end
else begin readln(b); writeln(query(1,b)); end;
end;
end.