[CodeVS1082] 线段树练习3(区间修改+询问区间和)

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