洛谷 P3373 (線段樹+雙標記)

在這裏插入圖片描述
題目鏈接

思路:先加後乘的操作是 a * c+b * c 先乘後加的操作是a * c+b * 1我們發現後者變成 * c就是前者,那麼我就定下優先級先乘後加,在區間乘的時候對加標記進行 * c操作push_down的時候也是先乘後加

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#include<iostream>

//#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define SIS std::ios::sync_with_stdio(false)
#define space putchar(' ')
#define enter putchar('\n')
#define lson root<<1
#define rson root<<1|1
typedef pair<int,int> PII;
const int mod=1e4+7;
const int N=1e5+10;
const int inf=0x7f7f7f7f;


ll gcd(ll a,ll b)
{
    return b==0?a:gcd(b,a%b);
}

ll lcm(ll a,ll b)
{
    return a*(b/gcd(a,b));
}

template <class T>
void read(T &x)
{
    char c;
    bool op = 0;
    while(c = getchar(), c < '0' || c > '9')
        if(c == '-')
            op = 1;
    x = c - '0';
    while(c = getchar(), c >= '0' && c <= '9')
        x = x * 10 + c - '0';
    if(op)
        x = -x;
}
template <class T>
void write(T x)
{
    if(x < 0)
        x = -x, putchar('-');
    if(x >= 10)
        write(x / 10);
    putchar('0' + x % 10);
}
int n,m,p;
struct node
{
    ll l,r;
    ll sum,mul=1,add=0;
}tre[N*4];
void push_up(int root)
{
    tre[root].sum=(tre[lson].sum+tre[rson].sum)%p;
    return;
}
void push_down(int root)
{
    int mid=tre[root].l+tre[root].r>>1;
    tre[lson].sum=(tre[lson].sum*tre[root].mul+tre[root].add*(mid-tre[root].l+1))%p;
    tre[rson].sum=(tre[rson].sum*tre[root].mul+tre[root].add*(tre[root].r-mid))%p;
    tre[lson].mul=(tre[lson].mul*tre[root].mul)%p;
    tre[rson].mul=(tre[rson].mul*tre[root].mul)%p;
    tre[lson].add=(tre[lson].add*tre[root].mul+tre[root].add)%p;
    tre[rson].add=(tre[rson].add*tre[root].mul+tre[root].add)%p;
    tre[root].add=0;
    tre[root].mul=1;return;

}

void build(int root,int l,int r)
{
    if(l==r)
    {
        tre[root].l=l;
        tre[root].r=r;
        cin>>tre[root].sum;return ;
    }
    tre[root].l=l;
    tre[root].r=r;
    int mid=l+r>>1;
    build(lson,l,mid);
    build(rson,mid+1,r);
    push_up(root);return;
}

void add(int root,int l,int r,ll k)
{
    if(l==tre[root].l&&r==tre[root].r)
    {
        tre[root].sum=(tre[root].sum+k*(tre[root].r-tre[root].l+1))%p;
        tre[root].add=(tre[root].add+k)%p;return ;
    }
    int mid=tre[root].l+tre[root].r>>1;
    push_down(root);
    if(r<=mid)
    {
        add(lson,l,r,k);
    }
    else
    {
        if(l>mid)
            add(rson,l,r,k);
        else
        {
            add(lson,l,mid,k);
            add(rson,mid+1,r,k);
        }
    }
    push_up(root);

}

void mul(int root,int l,int r,ll k)
{
       if(l==tre[root].l&&r==tre[root].r)
    {
         tre[root].sum=(tre[root].sum*k)%p;
         tre[root].mul=(tre[root].mul*k)%p;
         tre[root].add=(tre[root].add*k)%p;return ;
    }
    int mid=tre[root].l+tre[root].r>>1;
    push_down(root);
    if(r<=mid)
    {
        mul(lson,l,r,k);
    }
    else
    {
        if(l>mid)
        {
            mul(rson,l,r,k);
        }
        else
        {
            mul(lson,l,mid,k);
            mul(rson,mid+1,r,k);
        }

    }
    push_up(root);

}

ll query(int root,int l,int r)
{
    if(tre[root].l==l&&tre[root].r==r)
    {
        return tre[root].sum%p;
    }
    int mid=tre[root].l+tre[root].r>>1;
    push_down(root);
    if(r<=mid)
    {
        return query(lson,l,r);
    }
    else
    {
        if(l>mid)
            return query(rson,l,r);
        else
            return (query(lson,l,mid)+query(rson,mid+1,r))%p;
    }
}


int main()
{
    SIS;
    int t;
    cin>>n>>m>>p;
    build(1,1,n);
    while(m--)
    {
        int op;
        cin>>op;
        if(op==1)
        {
            int l,r;
            ll k;
            cin>>l>>r>>k;
            mul(1,l,r,k);
        }
        else if(op==2)
        {
            int l,r;
            ll k;
            cin>>l>>r>>k;
            add(1,l,r,k);
        }
        else
        {
            int l,r;
            cin>>l>>r;
            ll ans=query(1,l,r)%p;
            cout<<ans<<endl;

        }
    }




    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章