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.

     

 

我写的有些麻烦。。。

 

posted @ 2014-07-01 12:49  NanoApe  阅读(111)  评论(0编辑  收藏  举报
AmazingCounters.com