P3373 【模板】线段树 2

链接

题目描述 如题，已知一个数列，你需要进行下面三种操作： 将某区间每一个数乘上 xx 将某区间每一个数加上 xx 求出某区间每一个数的和 输入格式 第一行包含三个整数 n,m,pn,m,p，分别表示该数列数字的个数、操作的总个数和模数。 第二行包含 nn 个用空格分隔的整数，其中第 ii 个数字表示数列第 ii 项的初始值。 接下来 mm 行每行包含若干个整数，表示一个操作，具体如下： 操作 11： 格式：1 x y k 含义：将区间 [x,y][x,y] 内每个数乘上 kk 操作 22： 格式：2 x y k 含义：将区间 [x,y][x,y] 内每个数加上 kk 操作 33： 格式：3 x y 含义：输出区间 [x,y][x,y] 内每个数的和对 pp 取模所得的结果 输出格式 输出包含若干行整数，即为所有操作 33 的结果。 输入输出样例 输入 #1 复制 5 5 38 1 5 4 2 3 2 1 4 1 3 2 5 1 2 4 2 2 3 5 5 3 1 4 输出 #1 复制 17 2 说明/提示 【数据范围】 对于 30\%30% 的数据：n \le 8n≤8，m \le 10m≤10 对于 70\%70% 的数据：n \le 10^3 n≤10 3 ， m \le 10^4m≤10 4 对于 100\%100% 的数据： n \le 10^5n≤10 5 ， m \le 10^5m≤10 5 除样例外，p = 571373p=571373 数据已经过加强^_^） 样例说明：故输出应为 1717、22（ 40 \bmod 38 = 240mod38=2 ） 参考大佬代码，大概是先算乘法后算加法，数学不好略了。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define ll long long
const ll maxn=10e5+5;
struct tree
{
    ll sum,value,mul;
} trees[maxn];
void showtree(int u,int l,int r)
{
    if(l==r){
        printf("l:%d,r:%d,value:%lld,sum:%lld,mul%lld\n",l,r,trees[u].value,trees[u].sum,trees[u].mul);
        return ;
    }
     printf("l:%d,r:%d,value:%lld,sum:%lld,mul%lld\n",l,r,trees[u].value,trees[u].sum,trees[u].mul);
     int mid=(l+r)>>1;
     showtree(u<<1,l,mid);
     showtree(u<<1|1,mid+1,r);
}
ll p;
ll a[maxn];
void pushdown(ll u,ll l)
{
    trees[u<<1].value=((l-(l>>1))*trees[u].sum+trees[u<<1].value*trees[u].mul)%p;
    trees[u<<1|1].value=((l>>1)*trees[u].sum+trees[u<<1|1].value*trees[u].mul)%p;
    trees[u<<1|1].sum=(trees[u].sum+trees[u].mul*trees[u<<1|1].sum)%p;
    trees[u<<1].sum=(trees[u].sum+trees[u].mul*trees[u<<1].sum)%p;
    trees[u<<1|1].mul=(trees[u].mul*trees[u<<1|1].mul)%p;
    trees[u<<1].mul=(trees[u].mul*trees[u<<1].mul)%p;
    trees[u].sum=0;
    trees[u].mul=1;
}
void addupdate(ll l,ll r,ll left,ll right,ll u,ll k)
{
    if(l>=left&&r<=right)
    {
        trees[u].sum=(trees[u].sum+k)%p;
        trees[u].value=(trees[u].value+(r-l+1)*k)%p;
        return ;
    }
    int mid=(l+r)>>1;
    pushdown(u,r-l+1);
    if(left<=mid)
        addupdate(l,mid,left,right,u<<1,k);
    if(right>mid) addupdate(mid+1,r,left,right,u<<1|1,k);
    trees[u].value=(trees[u<<1].value+trees[u<<1|1].value)%p;
}
void mulupdate(ll l,ll r,ll left,ll right,ll u,ll k)
{
    if(l>=left&&r<=right)
    {
        trees[u].mul=(trees[u].mul*k)%p;
        trees[u].sum=(trees[u].sum*k)%p;
        trees[u].value=(trees[u].value*k)%p;
        return ;
    }
    int mid=(l+r)>>1;
    pushdown(u,r-l+1);
    if(left<=mid)
        mulupdate(l,mid,left,right,u<<1,k);
    if(right>mid) mulupdate(mid+1,r,left,right,u<<1|1,k);
    trees[u].value=(trees[u<<1].value+trees[u<<1|1].value)%p;
}
void buildtree(ll u,ll l,ll r)
{
    trees[u].sum=0;
    trees[u].mul=1;
    if(l==r)
    {
        trees[u].value=a[l]%p;
        return ;
    }
    ll mid=(l+r)>>1;
    buildtree(u<<1,l,mid);
    buildtree(u<<1|1,mid+1,r);
    trees[u].value=(trees[u<<1].value+trees[u<<1|1].value)%p;
}
void updatetree(ll l,ll r,ll left,ll right,ll u,ll k)
{
    if(l>=left&&r<=right)
    {
        trees[u].sum=(trees[u].sum+k)%p;
        trees[u].value=(trees[u].value+(r-l+1)*k)%p;
        return ;
    }
    ll mid=(l+r)>>1;
    if(trees[u].sum)
        pushdown(u,r-l+1);
    if(left<=mid)
    {
        updatetree(l,mid,left,right,u<<1,k);
    }
    if(right>mid)
    {
        updatetree(mid+1,r,left,right,u<<1|1,k);
    }
    trees[u].value=(trees[u<<1].value+trees[u<<1|1].value)%p;
}
ll querytree(ll l,ll r,ll left,ll right,ll u)
{
    if(l>=left&&r<=right)
        return trees[u].value%p;
    ll mid=(l+r)>>1;
    ll ans=0;
    pushdown(u,r-l+1);
    if(left<=mid)
        ans+=querytree(l,mid,left,right,u<<1);
    if(right>mid)
        ans+=querytree(mid+1,r,left,right,u<<1|1);
    return ans%p;
}
int main()
{
    ll n,q;
    scanf("%lld%lld%lld",&n,&q,&p);
    for(ll i=1; i<=n; i++)
        scanf("%lld",&a[i]);
    buildtree(1,1,n);
    for(int i=1;i<=q;i++)
    {
        int c;
        scanf("%d",&c);
        if(c==1)
        {
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            mulupdate(1,n,x,y,1,k);
        }
        else if(c==2)
        {
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            addupdate(1,n,x,y,1,k);
        }
        else
        {
            int x,y;
            scanf("%d%d",&x,&y);
            printf("%lld\n",querytree(1,n,x,y,1));
        }
    }

}