【P2023】[AHOI2009]维护序列 线段树

题目描述

老师交给小可可一个维护数列的任务，现在小可可希望你来帮他完成。 有长为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，按照它在输入中出现的顺序，依次输出一行一个整数表示询问结果。

输入输出样例

输入样例#1：

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

输出样例#1：

2
35
8

题解：

    好长时间没敲线段树板子了，调了好长时间，写几点注意吧

1. pushdown的时候不需要更新父节点的值
2. 先乘后加，pushdown的时候注意Mul变成1，Tag变成0；
3. 更新乘法标记的时候要把加法标记也乘上乘法标记（乘法···律.
4. pushdown的时候注意子节点的值得修改问题
5. 边写边%，

代码：

// luogu-judger-enable-o2
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define ll long long 
#define lson root<<1,left,mid
#define rson root<<1|1,mid+1,right 
#define Root ll root,ll left,ll right
const ll MAXN=1e6;
struct SEGTREE
{
    ll val,Tag,Mul;
}segTree[MAXN<<4];
ll val[MAXN],MOD;
ll n,m; 
using namespace std;
void Built(Root)
{
	segTree[root].Mul=1;
	segTree[root].Tag=0;
	if(left==right)
	{
		segTree[root].val=val[right];
		return ;
	}
	ll mid=(left+right)>>1;
	Built(lson);
	Built(rson);
	segTree[root].val=segTree[root<<1].val+segTree[root<<1|1].val; 
	segTree[root].val%=MOD; 
}
void pushdown(Root)
{
	ll mid=(left+right)>>1;
 	segTree[root<<1].Mul=(segTree[root<<1].Mul*segTree[root].Mul)%MOD;
	segTree[root<<1|1].Mul=(segTree[root<<1|1].Mul*segTree[root].Mul)%MOD;
	segTree[root<<1].Tag=(segTree[root].Tag+segTree[root<<1].Tag*segTree[root].Mul)%MOD;
	segTree[root<<1|1].Tag=(segTree[root].Tag+segTree[root<<1|1].Tag*segTree[root].Mul)%MOD;
	segTree[root<<1].val=((segTree[root<<1].val*segTree[root].Mul)%MOD+(segTree[root].Tag*(mid-left+1))%MOD)%MOD;
	segTree[root<<1|1].val=((segTree[root<<1|1].val*segTree[root].Mul)%MOD+(segTree[root].Tag*(right-mid))%MOD)%MOD;
	segTree[root].Mul=1;
	segTree[root].Tag=0; 	
} 
void Change_Mul(Root,ll qleft,ll qright,ll k)
{
	if(left>qright||right<qleft) return ;
	if(left>=qleft&&right<=qright)
	{
		segTree[root].Mul=(segTree[root].Mul*k)%MOD;
		segTree[root].Tag=(segTree[root].Tag*k)%MOD;
		segTree[root].val=(segTree[root].val*k)%MOD;
		return ;
	}
	pushdown(root,left,right);
	ll mid=(left+right)>>1;
	Change_Mul(lson,qleft,qright,k);
	Change_Mul(rson,qleft,qright,k);
	segTree[root].val=segTree[root<<1].val+segTree[root<<1|1].val;
	segTree[root].val%=MOD; 
}
void Change_Add(Root,ll qleft,ll qright,ll k)
{
	if(left>qright||right<qleft) return ;
	if(left>=qleft&&right<=qright)
	{
		segTree[root].Tag=(segTree[root].Tag+k)%MOD;
		segTree[root].val=(segTree[root].val+k*(right-left+1))%MOD;
		return ;
	}
	pushdown(root,left,right);
	ll mid=(left+right)>>1;
	Change_Add(lson,qleft,qright,k);
	Change_Add(rson,qleft,qright,k);
	segTree[root].val=segTree[root<<1].val+segTree[root<<1|1].val;
	segTree[root].val%=MOD;
}
ll Query(Root,ll qleft,ll qright)
{
	if(left>qright||right<qleft) return 0;
	if(left>=qleft&&right<=qright) return segTree[root].val%MOD;
	pushdown(root,left,right);
	ll mid=(left+right)>>1;
	return (Query(lson,qleft,qright)%MOD+Query(rson,qleft,qright)%MOD)%MOD;	
} 
int main()
{
    scanf("%lld%lld",&n,&MOD);
    for(int i=1;i<=n;i++) scanf("%lld",&val[i]);
    Built(1,1,n); 
    scanf("%lld",&m);
    for(int i=1;i<=m;i++)
    {
        ll opt,x,y,z;
        scanf("%lld",&opt);
        if(opt==1)
        {
            scanf("%lld%lld%lld",&x,&y,&z);
            Change_Mul(1,1,n,x,y,z);
        }
        if(opt==2)
        {
            scanf("%lld%lld%lld",&x,&y,&z);
            Change_Add(1,1,n,x,y,z);
        }
        if(opt==3)
        {
            scanf("%lld%lld",&x,&y);
            printf("%lld\n",Query(1,1,n,x,y));
        }
    }
    return 0;
}