线段树练习 3
题目描述 Description
给你N个数,有两种操作:
1:给区间[a,b]的所有数增加X
2:询问区间[a,b]的数的和。
输入描述 Input Description
第一行一个正整数n,接下来n行n个整数,
再接下来一个正整数Q,每行表示操作的个数,
如果第一个数是1,后接3个正整数,
表示在区间[a,b]内每个数增加X,如果是2,
表示操作2询问区间[a,b]的和是多少。
输出描述 Output Description
对于每个询问输出一行一个答案
样例输入 Sample Input
3
1
2
3
2
1 2 3 2
2 2 3
样例输出 Sample Output
9
数据范围及提示 Data Size & Hint
1<=n<=200000
1<=q<=200000
题解
- 这题超坑QAQ
- 读入n个数,并不是一行一个的….
- 读入的n个数并不能用longint存下…
var
w:array[0..1000005,1..4]of int64;
x:array[0..200005]of int64;
i,j,k:longint;
n,m,t:longint;
a,b,c:longint;
ans:int64;
procedure build(a,l,r:longint);
var mid:longint;
begin
w[a,1]:=l; w[a,2]:=r; w[a,4]:=0;
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);
w[a,3]:=w[a*2,3]+w[a*2+1,3];
end;
procedure update(a,l,r,c:longint);
var mid:longint;
begin
if (w[a,1]=l)and(w[a,2]=r)
then begin inc(w[a,4],c); exit; end;
inc(w[a,3],(r-l+1)*c);
mid:=(w[a,1]+w[a,2]) div 2;
if r<=mid then update(a*2,l,r,c) else
if l>=mid+1 then update(a*2+1,l,r,c)
else begin update(a*2,l,mid,c); update(a*2+1,mid+1,r,c); end;
end;
procedure pushdown(a:longint);
begin
inc(w[a,3],w[a,4]*(w[a,2]-w[a,1]+1));
inc(w[a*2,4],w[a,4]);
inc(w[a*2+1,4],w[a,4]);
w[a,4]:=0;
end;
function query(a,l,r:longint):int64;
var mid:longint;
begin
if w[a,4]<>0 then pushdown(a);
if (w[a,1]=l)and(w[a,2]=r)
then exit(w[a,3]);
mid:=(w[a,1]+w[a,2]) div 2;
if r<=mid then exit(query(a*2,l,r)) else
if l>=mid+1 then exit(query(a*2+1,l,r))
else exit(query(a*2,l,mid)+query(a*2+1,mid+1,r));
end;
begin
readln(n);
for i:=1 to n do
read(x[i]);
build(1,1,n);
readln(m);
for i:=1 to m do
begin
read(t);
if t=1
then begin readln(a,b,c); update(1,a,b,c); end
else begin readln(a,b); ans:=query(1,a,b); writeln(ans); end;
end;
end.