BZOJ-1798 维护序列

线段树。支持区间加、区间乘、区间查询和。

标记下移还有取模要注意。

• var

• n,p,q,i,s,t:longint;

• a:int64;

• num,n1,n2,n3:array[0..500000] of int64;

• procedure build(o,l,r:longint);

• var m,i:longint;

• begin

• m:=(l+r) div 2;

• if l=r then

• begin

• n1[o]:=num[l];

• n2[o]:=num[l];

• end

• else

• begin

• build(o*2,l,m);

• build(o*2+1,m+1,r);

• end;

• if o>1 then n1[o div 2]:=(n1[o div 2]+n1[o]) mod p;

• end;

• procedure add(o,l,r:longint);

• var m,i:longint;

• begin

• m:=(l+r) div 2;

• if l<>r then

• begin

• n1[o*2]:=(n1[o*2]*n3[o]+n2[o]*((m-l+1) mod p)) mod p;

• n2[o*2]:=(n2[o*2]*n3[o]+n2[o]) mod p;

• n3[o*2]:=(n3[o*2]*n3[o]) mod p;

• n1[o*2+1]:=(n1[o*2+1]*n3[o]+n2[o]*((r-m) mod p)) mod p;

• n2[o*2+1]:=(n2[o*2+1]*n3[o]+n2[o]) mod p;

• n3[o*2+1]:=(n3[o*2+1]*n3[o]) mod p;

• n2[o]:=0;n3[o]:=1;

• end;

• if (s<=l)and(r<=t) then

• begin

• n1[o]:=(n1[o]+a*((r-l+1) mod p)) mod p;

• n2[o]:=(n2[o]+a) mod p;

• end

• else

• begin

• if s<m+1 then add(o*2,l,m);

• if m<t then add(o*2+1,m+1,r);

• n1[o]:=(n1[o*2]+n1[o*2+1]) mod p;

• end;

• end;

• procedure che(o,l,r:longint);

• var m,i:longint;

• begin

• m:=(l+r) div 2;

• if l<>r then

• begin

• n1[o*2]:=(n1[o*2]*n3[o]+n2[o]*((m-l+1) mod p)) mod p;

• n2[o*2]:=(n2[o*2]*n3[o]+n2[o]) mod p;

• n3[o*2]:=(n3[o*2]*n3[o]) mod p;

• n1[o*2+1]:=(n1[o*2+1]*n3[o]+n2[o]*((r-m) mod p)) mod p;

• n2[o*2+1]:=(n2[o*2+1]*n3[o]+n2[o]) mod p;

• n3[o*2+1]:=(n3[o*2+1]*n3[o]) mod p;

• n2[o]:=0;n3[o]:=1;

• end;

• if (s<=l)and(r<=t) then

• begin

• n1[o]:=(n1[o]*a) mod p;

• n2[o]:=(n2[o]*a) mod p;

• n3[o]:=(n3[o]*a) mod p;

• end

• else

• begin

• if s<m+1 then che(o*2,l,m);

• if m<t then che(o*2+1,m+1,r);

• n1[o]:=(n1[o*2]+n1[o*2+1]) mod p;

• end;

• end;

• function que(o,l,r:longint):int64;

• var m,i:longint;

• begin

• m:=(l+r) div 2;

• que:=0;

• if l<>r then

• begin

• n1[o*2]:=(n1[o*2]*n3[o]+n2[o]*((m-l+1) mod p)) mod p;

• n2[o*2]:=(n2[o*2]*n3[o]+n2[o]) mod p;

• n3[o*2]:=(n3[o*2]*n3[o]) mod p;

• n1[o*2+1]:=(n1[o*2+1]*n3[o]+n2[o]*((r-m) mod p)) mod p;

• n2[o*2+1]:=(n2[o*2+1]*n3[o]+n2[o]) mod p;

• n3[o*2+1]:=(n3[o*2+1]*n3[o]) mod p;

• n2[o]:=0;n3[o]:=1;

• end;

• if (s<=l)and(r<=t) then que:=n1[o]

• else

• begin

• if s<m+1 then que:=(que+que(o*2,l,m)) mod p;

• if m<t then que:=(que+que(o*2+1,m+1,r)) mod p;

• end;

• end;

• begin

• read(n,p);

• for i:=1 to n do read(num[i]);

• for i:=1 to n do num[i]:=num[i] mod p;

• for i:=1 to 3*n do n3[i]:=1;

• build(1,1,n);

• read(q);

• for i:=1 to q do

• begin

• read(a);

• case a of

• 1:begin

• read(s,t,a);

• a:=a mod p;

• che(1,1,n);

• end;

• 2:begin

• read(s,t,a);

• a:=a mod p;

• add(1,1,n);

• end;

• 3:begin

• read(s,t);

• writeln(que(1,1,n));

• end;

• end;

• end;

• end.

我写的有些麻烦。。。