UESTC - 1597 An easy problem C （线段树）

An easy problem C

N个数排成一列，有三种操作。1.给一段区间内的每个数乘上一个非负整数。2.给一段区间内的每个数加上一个非负整数.3.询问一段区间的和模上P的值。

Input

第一行两个整数N(1≤N≤100000)表示数的个数,P（1≤P≤1000000000）表示模的值。接下来一行N个整数ai(0≤ai≤1000000000),接下来一行一个整数M(1≤M≤100000)表示操作数量，接下来M行每行描述一个操作。第一种操作描述：1 L R C（0≤C≤1000000000）,表示把L到R这段区间每个数乘上一个C。第二种操作描述：2 L R C（0≤C≤1000000000）,表示把L到R这段区间每个数加上一个C。第三种操作3 L R 表示询问L到R这段区间内的数的和模上P的值。

Output

对面每个询问，输出对应的答案，每个询问占一行。

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

思路：hdu 4578弱化版，但是这道题应该注意的是mul[i]要初始为1

其他的依旧在Pushdown的时候，先更新mul，再更新add

代码:

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

typedef long long LL;
const LL maxn=1e5+10;
struct Segtree
{
    LL le,ri;
    LL val,add,mul;
    LL mid()
    {
        return (le+ri)>>1;
    }
} tree[maxn<<2];
LL a[maxn];
LL n,mod;

void fun_add(LL rt,LL num)
{
    tree[rt].add=(tree[rt].add+num)%mod;
    LL len=tree[rt].ri-tree[rt].le+1;
    tree[rt].val=(tree[rt].val+(num*len)%mod)%mod;
}

void fun_mul(LL rt,LL num)
{
    tree[rt].mul=(tree[rt].mul*num)%mod;
    tree[rt].add=(tree[rt].add*num)%mod;
    tree[rt].val=(tree[rt].val*num)%mod;
}

void Pushdown(LL rt)
{
    if(tree[rt].mul!=1)
    {
        fun_mul(rt<<1,tree[rt].mul);
        fun_mul(rt<<1|1,tree[rt].mul);
        tree[rt].mul=1;
    }
    if(tree[rt].add)
    {
        fun_add(rt<<1,tree[rt].add);
        fun_add(rt<<1|1,tree[rt].add);
        tree[rt].add=0;
    }
}

void Pushup(LL rt)
{
    tree[rt].val=(tree[rt<<1].val+tree[rt<<1|1].val)%mod;
}

void Build(LL rt,LL le,LL ri)
{
    tree[rt].le=le,tree[rt].ri=ri;
    tree[rt].add=0;
    tree[rt].mul=1;
    if(le==ri)
    {
        tree[rt].val=a[le];
        return ;
    }
    LL mid=tree[rt].mid();
    Build(rt<<1,le,mid);
    Build(rt<<1|1,mid+1,ri);
    Pushup(rt);
}

void Update(LL rt,LL le,LL ri,LL num,LL op)
{
    if(le<=tree[rt].le&&tree[rt].ri<=ri)
    {
        if(op==1)
            fun_mul(rt,num);
        else
            fun_add(rt,num);
        return ;
    }
    Pushdown(rt);
    LL mid=tree[rt].mid();
    if(le<=mid)
        Update(rt<<1,le,ri,num,op);
    if(ri>mid)
        Update(rt<<1|1,le,ri,num,op);
    Pushup(rt);
}

LL Query(LL rt,LL le,LL ri)
{
    if(le<=tree[rt].le&&tree[rt].ri<=ri)
        return tree[rt].val;
    Pushdown(rt);
    LL mid=tree[rt].mid();
    LL ans=0;
    if(le<=mid)
        ans=(ans+Query(rt<<1,le,ri))%mod;
    if(ri>mid)
        ans=(ans+Query(rt<<1|1,le,ri))%mod;
    Pushup(rt);
    return ans;
}

int main()
{
    scanf("%lld%lld",&n,&mod);
    for(LL i=1; i<=n; ++i)
        scanf("%lld",&a[i]);
    Build(1,1,n);
    LL op,x,y,m;
    LL c;
    scanf("%lld",&m);
    while(m--)
    {
        scanf("%lld",&op);
        if(op<=2)
        {
            scanf("%lld%lld%lld",&x,&y,&c);
            Update(1,x,y,c,op);
        }
        else
        {
            scanf("%lld%lld",&x,&y);
            printf("%lld\n",Query(1,x,y));
        }
    }
    return 0;
}