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.

我写的有些麻烦。。。