p2023&bzoj1798 维护序列

传送门（洛谷）

传送门（bzoj）

题目

老师交给小可可一个维护数列的任务，现在小可可希望你来帮他完成。 有长为N的数列，不妨设为a1,a2,…,aN 。有如下三种操作形式： (1)把数列中的一段数全部乘一个值; (2)把数列中的一段数全部加一个值; (3)询问数列中的一段数的和，由于答案可能很大，你只需输出这个数模P的值。

输入格式：

第一行两个整数N和P(1≤P≤1000000000）。
第二行含有N个非负整数,从左到右依次为a1,a2,…,aN, (0≤ai≤1000000000,1≤i≤N)。
第三行有一个整数M，表示操作总数。
从第四行开始每行描述一个操作，输入的操作有以下三种形式：
操作1：“1 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai×c (1≤t≤g≤N,0≤c≤1000000000)。
操作2：“2 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai+c (1≤t≤g≤N,0≤c≤1000000000)。
操作3：“3 t g”(不含双引号)。询问所有满足t≤i≤g的ai的和模P的值 (1≤t≤g≤N)。 同一行相邻两数之间用一个空格隔开，每行开头和末尾没有多余空格。

输出格式：

对每个操作3，按照它在输入中出现的顺序，依次输出一行一个整数表示询问结果。

分析

参见洛谷线段树模板2，lazy标记记录两个值mul和add。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
long long n,m,mod;
struct node {
      long long sum,add,mul;
}d[];
inline void build(long long le,long long ri,long long pos,long long wh,long long k){
      d[wh].sum+=k;
      d[wh].mul=;
      if(le==ri)return;
      long long mid=(le+ri)>>;
      if(mid>=pos)build(le,mid,pos,wh<<,k);
        else build(mid+,ri,pos,(wh<<)|,k);
}
inline void udm(long long le,long long ri,long long x,long long y,long long wh,long long k){
      if(le>=x&&ri<=y){
          d[wh].mul=d[wh].mul*k%mod;
          d[wh].add=d[wh].add*k%mod;
          d[wh].sum=d[wh].sum*k%mod;
          return;
      }
      long long mid=(le+ri)>>;
      long long t=d[wh].mul,s=d[wh].add;
      d[wh].mul=,d[wh].add=;
      d[wh<<].mul=d[wh<<].mul*t%mod;
      d[wh<<].add=(d[wh<<].add*t%mod+s)%mod;
      d[wh<<].sum=(d[wh<<].sum*t%mod+(mid-le+)*s%mod)%mod;
      d[(wh<<)|].mul=d[(wh<<)|].mul*t%mod;
      d[(wh<<)|].add=(d[(wh<<)|].add*t%mod+s)%mod;
      d[(wh<<)|].sum=(d[(wh<<)|].sum*t%mod+(ri-mid)*s%mod)%mod;
      if(mid>=x)udm(le,mid,x,y,wh<<,k);
      if(mid<y)udm(mid+,ri,x,y,(wh<<)|,k);
      d[wh].sum=(d[wh<<].sum+d[(wh<<)|].sum)%mod;
}
inline void uda(long long le,long long ri,long long x,long long y,long long wh,long long k){
      if(le>=x&&ri<=y){
          d[wh].add=(d[wh].add+k)%mod;
          d[wh].sum=(d[wh].sum+(ri-le+)*k%mod)%mod;
          return;
      }
      long long mid=(le+ri)>>;
      long long t=d[wh].mul,s=d[wh].add;
      d[wh].mul=,d[wh].add=;
      d[wh<<].mul=d[wh<<].mul*t%mod;
      d[wh<<].add=(d[wh<<].add*t%mod+s)%mod;
      d[wh<<].sum=(d[wh<<].sum*t%mod+(mid-le+)*s%mod)%mod;
      d[(wh<<)|].mul=d[(wh<<)|].mul*t%mod;
      d[(wh<<)|].add=(d[(wh<<)|].add*t%mod+s)%mod;
      d[(wh<<)|].sum=(d[(wh<<)|].sum*t%mod+(ri-mid)*s%mod)%mod;
      if(mid>=x)uda(le,mid,x,y,wh<<,k);
      if(mid<y)uda(mid+,ri,x,y,(wh<<)|,k);
      d[wh].sum=(d[wh<<].sum+d[(wh<<)|].sum)%mod;
}
inline long long que(long long le,long long ri,long long x,long long y,long long wh){
      if(le>=x&&ri<=y)return d[wh].sum%mod;
      long long mid=(le+ri)>>,ans=;
      long long t=d[wh].mul,s=d[wh].add;
      d[wh].mul=,d[wh].add=;
      d[wh<<].mul=d[wh<<].mul*t%mod;
      d[wh<<].add=(d[wh<<].add*t%mod+s)%mod;
      d[wh<<].sum=(d[wh<<].sum*t%mod+(mid-le+)*s%mod)%mod;
      d[(wh<<)|].mul=d[(wh<<)|].mul*t%mod;
      d[(wh<<)|].add=(d[(wh<<)|].add*t%mod+s)%mod;
      d[(wh<<)|].sum=(d[(wh<<)|].sum*t%mod+(ri-mid)*s%mod)%mod;
      if(mid>=x)ans=(ans+que(le,mid,x,y,wh<<))%mod;
      if(mid<y)ans=(ans+que(mid+,ri,x,y,(wh<<)|))%mod;
      d[wh].sum=(d[wh<<].sum+d[(wh<<)|].sum)%mod;
      return ans;
}
int main()
{     long long i,j,k,x,y,z;
      scanf("%lld%lld",&n,&mod);
      for(i=;i<=n;i++){
          scanf("%lld",&x);
        build(,n,i,,x);
      }
      scanf("%lld",&m);
      for(i=;i<=m;i++){
          scanf("%lld",&k);
          if(k==){
              scanf("%lld%lld%lld",&x,&y,&z);
              udm(,n,x,y,,z);
          }else if(k==){
              scanf("%lld%lld%lld",&x,&y,&z);
              uda(,n,x,y,,z);
          }else {
              scanf("%lld%lld",&x,&y);
              printf("%lld\n",que(,n,x,y,));
          }
      }
      return ;
}