Description
老师交给小可可一个维护数列的任务,现在小可可希望你来帮他完成。 有长为N的数列,不妨设为a1,a2,…,aN 。有如下三种操作形式: (1)把数列中的一段数全部乘一个值; (2)把数列中的一段数全部加一个值; (3)询问数列中的一段数的和,由于答案可能很大,你只需输出这个数模P的值。
Input
第一行两个整数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)。 同一行相邻两数之间用一个空格隔开,每行开头和末尾没有多余空格。
Output
对每个操作3,按照它在输入中出现的顺序,依次输出一行一个整数表示询问结果。
Sample Input
7 43
1 2 3 4 5 6 7
5
1 2 5 5
3 2 4
2 3 7 9
3 1 3
3 4 7
Sample Output
2
35
8
题目分析:
一个裸的线段树双标记
假设sum是节点保存的和 面对mul,add双标记:
(sum+add*(a[num].y-a[num].x+1))*mul+add*(a[num].y-a[num].x+1) = mul*sum+(add*mul+add)*(a[num].y-a[num].x+1)
这样 我们可以在更新mul标记时直接将add标记乘上要乘的值
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m;
long long mod;
long long c[100100];
struct your
{
long long mul;
long long add;
long long sum;
int x,y;
}a[400400];
void build(int dx,int dy,int num)
{
a[num].x=dx,a[num].y=dy;
a[num].add=0,a[num].mul=1;
if(dx==dy)
{
a[num].sum=c[dx]%mod;
return ;
}
int mid=(dx+dy)>>1;
build(dx,mid,num<<1);
build(mid+1,dy,num<<1|1);
a[num].sum=(a[num<<1].sum+a[num<<1|1].sum)%mod;
}
void push_down(int num)
{
a[num<<1].add=(a[num<<1].add*a[num].mul+a[num].add)%mod;
a[num<<1|1].add=(a[num<<1|1].add*a[num].mul+a[num].add)%mod;
a[num<<1].mul=(a[num<<1].mul*a[num].mul)%mod;
a[num<<1|1].mul=(a[num<<1|1].mul*a[num].mul)%mod;
a[num<<1].sum=(a[num<<1].sum*a[num].mul+a[num].add*(a[num<<1].y-a[num<<1].x+1))%mod;
a[num<<1|1].sum=(a[num<<1|1].sum*a[num].mul+a[num].add*(a[num<<1|1].y-a[num<<1|1].x+1))%mod;
a[num].add=0;
a[num].mul=1;
return ;
}
void update_mul(int dx,int dy,long long val,int num)
{
if(a[num].x==dx&&a[num].y==dy)
{
a[num].mul=(a[num].mul*val)%mod;
a[num].add=(a[num].add*val)%mod;
a[num].sum=(a[num].sum*val)%mod;
return ;
}
if(a[num].mul!=1||a[num].add!=0)
push_down(num);
int mid=(a[num].x+a[num].y)>>1;
if(dx>mid) update_mul(dx,dy,val,num<<1|1);
else if(dy<=mid) update_mul(dx,dy,val,num<<1);
else
{
update_mul(dx,mid,val,num<<1);
update_mul(mid+1,dy,val,num<<1|1);
}
a[num].sum=(a[num<<1].sum+a[num<<1|1].sum)%mod;
return ;
}
void update_add(int dx,int dy,long long val,int num)
{
if(a[num].x==dx&&a[num].y==dy)
{
a[num].add=(a[num].add+val)%mod;
a[num].sum=(a[num].sum+val*(a[num].y-a[num].x+1))%mod;
return ;
}
if(a[num].mul!=1||a[num].add!=0)
push_down(num);
int mid=(a[num].x+a[num].y)>>1;
if(dx>mid) update_add(dx,dy,val,num<<1|1);
else if(dy<=mid) update_add(dx,dy,val,num<<1);
else
{
update_add(dx,mid,val,num<<1);
update_add(mid+1,dy,val,num<<1|1);
}
a[num].sum=(a[num<<1].sum+a[num<<1|1].sum)%mod;
return ;
}
long long ask(int dx,int dy,int num)
{
if(a[num].x==dx&&a[num].y==dy)
return a[num].sum%mod;
if(a[num].mul!=1||a[num].add!=0)
push_down(num);
int mid=(a[num].x+a[num].y)>>1;
if(dx>mid) return ask(dx,dy,num<<1|1)%mod;
if(dy<=mid) return ask(dx,dy,num<<1)%mod;
return (ask(dx,mid,num<<1)+ask(mid+1,dy,num<<1|1))%mod;
}
int main()
{
scanf("%d%lld",&n,&mod);
for(int i=1;i<=n;i++)
scanf("%lld",&c[i]);
build(1,n,1);
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
int tmp;
scanf("%d",&tmp);
if(tmp==1)
{
int x,y;
long long c;
scanf("%d%d%lld",&x,&y,&c);
update_mul(x,y,c,1);
}
if(tmp==2)
{
int x,y;
long long c;
scanf("%d%d%lld",&x,&y,&c);
update_add(x,y,c,1);
}
if(tmp==3)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%lld\n",ask(x,y,1)%mod);
}
}
}