线段树(codevs1082)

时间:2023-03-09 08:21:57
线段树(codevs1082)
type jd=record
z,y,lc,rc,sum,toadd:int64;
end; var
tree:array[..] of jd;
qzh:array[..] of int64;
x:array[..] of int64;
n,m,a,b,k,ans,tot,t:int64;
i,j:longint; function min(a,b:longint):longint;
begin
if a<b then exit(a) else exit(b);
end; function max(a,b:longint):longint;
begin
if a>b then exit(a) else exit(b);
end; procedure maketree(a,b:longint);
var
mid,t:longint;
begin
inc(tot);
t:=tot;
tree[t].z:=a;tree[t].y:=b;
tree[t].sum:=qzh[b]-qzh[a-];
mid:=(a+b) div ;
if a<b then
begin
tree[t].lc:=tot+;
maketree(a,mid);
tree[t].rc:=tot+;
maketree(mid+,b);
end;
end; procedure add(po,a,b,k:longint);
var
mid:longint;
begin
mid:=(tree[po].z+tree[po].y) div ; tree[po].sum:=tree[po].sum+k*(min(b,tree[po].y)-max(a,tree[po].z)+)+tree[po].toadd*(tree[po].y-tree[po].z+); if tree[po].toadd<> then
begin
tree[tree[po].lc].toadd:=tree[tree[po].lc].toadd+tree[po].toadd;
tree[tree[po].rc].toadd:=tree[tree[po].rc].toadd+tree[po].toadd;
tree[po].toadd:=;
end; if (a=tree[po].z)and(b=tree[po].y) then
begin
tree[tree[po].lc].toadd:=tree[tree[po].lc].toadd+k;
tree[tree[po].rc].toadd:=tree[tree[po].rc].toadd+k;
end else
begin
if a<=mid then add(tree[po].lc,a,min(b,mid),k);
if b>mid then add(tree[po].rc,max(a,mid+),b,k);
end;
end; procedure getans(po,a,b:longint);
var
mid:longint;
begin
if tree[po].toadd<> then
begin
tree[po].sum:=tree[po].sum+tree[po].toadd*(tree[po].y-tree[po].z+);
tree[tree[po].lc].toadd:=tree[tree[po].lc].toadd+tree[po].toadd;
tree[tree[po].rc].toadd:=tree[tree[po].rc].toadd+tree[po].toadd;
tree[po].toadd:=;
end; mid:=(tree[po].z+tree[po].y) div ;
if (a=tree[po].z) and (b=tree[po].y) then
ans:=ans+tree[po].sum else
begin
if a<=mid then getans(tree[po].lc,a,min(b,mid));
if b>mid then getans(tree[po].rc,max(a,mid+),b);
end;
end; begin readln(n);
for i:= to n do
begin
read(x[i]);
qzh[i]:=qzh[i-]+x[i];
end; maketree(,n); read(m);
for i:= to m do
begin
read(t);
if t= then
begin
read(a,b,k);
add(,a,b,k);
end;
if t= then
begin
read(a,b);
ans:=;
getans(,a,b);
writeln(ans);
end;
end; end.