线段树 （P3373 【模板】线段树 2）

题目描述

如题，已知一个数列，你需要进行下面三种操作：

1.将某区间每一个数乘上x

2.将某区间每一个数加上x

3.求出某区间每一个数的和

输入输出格式

输入格式：

 

第一行包含三个整数N、M、P，分别表示该数列数字的个数、操作的总个数和模数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作1： 格式：1 x y k 含义：将区间[x,y]内每个数乘上k

操作2： 格式：2 x y k 含义：将区间[x,y]内每个数加上k

操作3： 格式：3 x y 含义：输出区间[x,y]内每个数的和对P取模所得的结果

 

输出格式：

 

输出包含若干行整数，即为所有操作3的结果。

 

输入输出样例

输入样例#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

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=1000，M<=10000

对于100%的数据：N<=100000，M<=100000

（数据已经过加强^_^）

样例说明：

故输出应为17、2（40 mod 38=2）

犯了2个错误，一个是把区间更新的 乘与加 分开来写（没想到延迟更新时候会互相影响）QAQ,第二个错误是updata的时候 理所当然只传入了当前区间，没有考虑当前区间与原来要更新的区间的关系，改了差不多1天 TvT.

代码：

#include <math.h>
#include <algorithm>
#include <iostream>
#include <queue>
#include <cstdio>
#include <list>
#include <deque>
#include <set>
#include <vector>
#include <map>
#include <cstring>
#include <iomanip>
#define max(a,b) (a>b?a:b)
#define LL long long
#define ll long long
using namespace std;
const int MAXN=4e5+7;
struct node
{
    LL l,r;
    LL val;
    LL addmark;//延迟标记
    LL Addmark;
}tree[MAXN];
LL p;
void build(int root,LL a[],LL l,LL r)
{
    tree[root].addmark=0;
    tree[root].Addmark=1;
    tree[root].l=l;
    tree[root].r=r;
    if(l==r)
    {
        tree[root].val=a[l];
        return;
    }
    else
    {
    LL mid=(l+r)/2;
    build(root*2,a,l,mid);
    build(root*2+1,a,mid+1,r);
    tree[root].val=tree[root*2].val+tree[root*2+1].val;
    }
    tree[root].val%=p;
    return;
}
/*
传递延迟标志 ,更新值
*/
void pushdown(int root)
{
    int l=tree[root].l;
    int r=tree[root].r;
    int mid=(tree[root].l+tree[root].r)/2;
    tree[root*2].val=(tree[root*2].val*tree[root].Addmark+tree[root].addmark*(mid-l+1))%p;
    tree[root*2+1].val=(tree[root*2+1].val*tree[root].Addmark+tree[root].addmark*(r-mid))%p;

    tree[root*2].Addmark=(tree[root*2].Addmark*tree[root].Addmark)%p;
    tree[root*2+1].Addmark=(tree[root*2+1].Addmark*tree[root].Addmark)%p;
    tree[root*2].addmark=(tree[root*2].addmark*tree[root].Addmark+tree[root].addmark)%p;
    tree[root*2+1].addmark=(tree[root*2+1].addmark*tree[root].Addmark+tree[root].addmark)%p;

    tree[root].Addmark=1;
    tree[root].addmark=0;
    return;
}
void Update1(int root,LL l,LL r,LL val)
{
  if(tree[root].l>r||tree[root].r<l)
        return;
  if(tree[root].l>=l&&tree[root].r<=r)
  {
      tree[root].Addmark=(tree[root].Addmark*val)%p;
      tree[root].val=(tree[root].val*val)%p;
      tree[root].addmark=(tree[root].addmark*val)%p;
      return;
  }
    pushdown(root);
    LL mid=(tree[root].l+tree[root].r)/2;
    //PUSH(root);//延迟标记 向下传递
    if(r<=mid)
    Update1(root*2,l, r, val);
    else if(l>mid)
    Update1(root*2+1, l, r, val);
    else
    {
    Update1(root*2,l, mid, val);
    Update1(root*2+1, mid+1, r, val);
    }
    //Update1(root*2,l,mid,val);
   // Update1(root*2+1,mid+1,r,val);
    tree[root].val=(tree[root*2].val+tree[root*2+1].val)%p;
return;
}

void Update(int root,int l,int r,int val)
{
  if(tree[root].l>r||tree[root].r<l)
        return;
  if(tree[root].l>=l&&tree[root].r<=r)
  {
      tree[root].addmark=(tree[root].addmark+val)%p;
      tree[root].val=(tree[root].val+val*(r-l+1))%p;
      return;
  }
    pushdown(root);//延迟标记 向下传递

    LL mid=(tree[root].l+tree[root].r)/2;
    if(r<=mid)
    Update(root*2,l, r, val);
    else if(l>mid)
    Update(root*2+1, l, r, val);
    else
    {
    Update(root*2,l, mid, val);
    Update(root*2+1, mid+1, r, val);
    }
   // Update(root*2,l,mid,val);
   // Update(root*2+1,mid+1,r,val);
    tree[root].val=(tree[root*2].val+tree[root*2+1].val)%p;
return;
}
LL Query(int root,int l,int r)
{
    LL sum=0;
//    Push(root);
    if(tree[root].l==l&&tree[root].r==r)
    {
       return tree[root].val;
    }
    LL mid=(tree[root].l+tree[root].r)/2;
    //PUSH(root);
    //Push(root);
    pushdown(root);
    if(r<=mid)
    {
        sum+=Query(root*2,l,r)%p;
    }
    else
    {  //mid r
        if(l>=mid+1)
        {
            sum+=Query(root*2+1,l,r)%p;
            sum%=p;
        }
        else
        {
            sum+=(Query(root*2,l,mid)+Query(root*2+1,mid+1,r))%p;
            sum%=p;
        }
    }
    return sum%p;
}
LL num[100007]={0};
int main()
{
    ios::sync_with_stdio(false);
    LL n,m;

    cin>>n>>m>>p;
    for(int i=1;i<=n;i++)
        cin>>num[i];
    build(1,num,1,n);
    while(m--)
    {
        int a;
        cin>>a;
        if(a==1)
        {
            LL x,y,k;
            cin>>x>>y>>k;
            Update1(1,x,y,k);
            //cout<<Query(1,x,y)<<endl;
        }
        else if(a==2)
        {
            LL l,r,val;
            cin>>l>>r>>val;
            Update(1,l,r,val);
        }
        else if(a==3)
        {
            int l,r;
            cin>>l>>r;
            cout<<Query(1,l,r)<<endl;

        }
       // gettree(n);

    }
return 0;
}

/*
void gettree(int n)
{
    for(int i=0;i<=n;i++)
    {
        cout<<"i= "<<i<<endl;
        cout<<tree[i].l<<" "<<tree[i].r<<" "<<tree[i].val<<endl;
        cout<<tree[i].addmark<<endl;
    }
}
*/
/*
/*
void Push(int root)
{
    if(tree[root].addmark!=0)
    {
        //PUSH(root);
        //可能左右节点被多次标记
        tree[root*2].addmark+=tree[root].addmark;
        tree[root*2+1].addmark+=tree[root].addmark;

        tree[root*2].val+=tree[root].addmark;
        tree[root*2+1].val+=tree[root].addmark;
        //tree[root*2].val%=p;
        //tree[root*2+1].val%=p;
        tree[root].addmark=0;
    }
}
void PUSH(int root)
{
   if(tree[root].Addmark!=1)
    {
        //Push(root);
        //可能左右节点被多次标记
        tree[root*2].Addmark=(tree[root*2].Addmark*tree[root].Addmark)%p;
        tree[root*2+1].Addmark=(tree[root*2+1].Addmark*tree[root].Addmark)%p;

        tree[root*2].val=(tree[root*2].val*tree[root].Addmark)%p;
        tree[root*2+1].val=(tree[root*2+1].val*tree[root].Addmark)%p;
        //tree[root*2+1].val%=p;
       // tree[root*2].val%=p;
        tree[root].Addmark=1;
    }
}*/