CDOJ1597-線段樹好題(2017 UESTC Training for Data Structures C)

傳送門:CDOJ1597


題目大意:

給你一個長度爲n的序列,m次操作,三種操作:

1.給一段區間內的每個數乘上一個非負整數。

2.給一段區間內的每個數加上一個非負整數.

3.詢問一段區間的和模上P的值。


題目思路:

首先我們很好想到用線段樹來維護這個區間的和,但是這裏多了個乘法,加法我們很好處理

其實想一想乘法的話我們可以另外加個乘法的懶惰數組,這樣在更新時照樣跟新就是,這裏的問題

就是加法和乘法的更新順序,我們可以先確定更新順序,我們先更新乘法,在更新加法,我們知道

更新乘法是如果前面有加的,我們必須先加上在乘,而這裏我們可以讓加的數組加上它本身的乘以

要乘的數,這樣我們就可以毫不猶豫按照我們的順序更新


AC代碼:

#include<bits/stdc++.h>
using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define ll long long
const int maxn = 1e5+100;
ll p;
int n;
ll Tree[maxn<<2],cadd[maxn<<2],cmul[maxn<<2];
void Build(int l,int r,int rt)
{
    cadd[rt] = 0;
    cmul[rt] = 1;

    if(l==r)
    {
        scanf("%lld",&Tree[rt]);
        return ;
    }
    int mid = (l+r)>>1;

    Build(lson);
    Build(rson);
    Tree[rt] = Tree[rt<<1]+Tree[rt<<1|1];
    Tree[rt]%=p;
}

void pushdown(int rt,ll m)
{
    
    //先讓加的數組加上他要乘的倍數
    cadd[rt<<1]=(cadd[rt<<1]*cmul[rt]+cadd[rt])%p;
    cadd[rt<<1|1]=(cadd[rt<<1|1]*cmul[rt]+cadd[rt])%p;

    
    //先更新乘法
    cmul[rt<<1] *= cmul[rt];
    cmul[rt<<1|1] *= cmul[rt];

    cmul[rt<<1] %= p;
    cmul[rt<<1|1] %= p;

    //後更新加法
    Tree[rt<<1]=(Tree[rt<<1]*cmul[rt]+(m-m/2ll)*cadd[rt])%p;
    Tree[rt<<1|1]=(Tree[rt<<1|1]*cmul[rt]+(m/2ll)*cadd[rt])%p;

    cadd[rt] = 0;
    cmul[rt] = 1;

}

void updata(int L,int R,int l,int r,int rt ,ll c,int op)
{
    if(L<=l&&r<=R)
    {
        if(op==1)
        {
            cadd[rt] = cadd[rt]*c;
            cadd[rt]%=p;
            cmul[rt]*=c;
            cmul[rt]%=p;
            Tree[rt]*=c;
            Tree[rt]%=p;
        }
        else
        {
            cadd[rt]+=c;
            cadd[rt]%=p;
            Tree[rt]+=c*(r-l+1ll);
            Tree[rt]%=p;
        }

        return ;
    }

    pushdown(rt,r-l+1ll);

    int mid = (l+r)>>1;
    if(L<=mid)
    {
        updata(L,R,lson,c,op);
    }
    if(R>mid)
    {
        updata(L,R,rson,c,op);
    }

    Tree[rt] = (Tree[rt<<1]+Tree[rt<<1|1])%p;
}

ll quary(int L,int R,int l,int r,int rt)
{

    if(L<=l&&r<=R)return Tree[rt]%p;
    pushdown(rt,r-l+1ll);
    int mid = (l+r)>>1;
    ll res = 0;
    if(L<=mid)res+=quary(L,R,lson);
    if(R>mid)res+=quary(L,R,rson);
    return res%p;
}

int main()
{
    cin>>n>>p;
    Build(1,n,1);
    int m;
    cin>>m;
    while(m--)
    {
        int id,a,b;
        ll c;
        scanf("%d",&id);
        if(id==1)
        {
            scanf("%d%d%lld",&a,&b,&c);
            updata(a,b,1,n,1,c,1);
        }
        else if(id==2)
        {
            scanf("%d%d%lld",&a,&b,&c);
            updata(a,b,1,n,1,c,2);
        }
        else
        {
            scanf("%d%d",&a,&b);
            printf("%lld\n",quary(a,b,1,n,1));
        }
    }

    return 0;
}








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