[bzoj 1798--AHOI2009&bzoj 5039--JSOI2014]维护序列&序列维护

124 篇文章 2 订阅
14 篇文章 0 订阅

老师交给小可可一个维护数列的任务,现在小可可希望你来帮他完成。 有长为N的数列,不妨设为a1,a2,…,aN 。有如下三种操作形式:
(1)把数列中的一段数全部乘一个值;
(2)把数列中的一段数全部加一个值;
(3)询问数列中的一段数的和,由于答案可能很大,你只需输出这个数模P的值。

这道题一眼线段树,只不过是要维护两种标记。
不过仔细想一想也挺简单的。
设先乘后加,(m1,m2),m1为要乘的数,m2为要加的数。
那乘了个数k之后,(m1,m2)就变成(m1*k,m2*k)
加了个数k之后,(m1,m2)就变成(m1,m2+k)。
那之后只要按照上面方法维护就解决了。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
using namespace std;
struct node
{
    int l,r,lc,rc;
    long long c,jf,cf;
}tr[210000];int trlen;
int n,m;long long mod,a[110000];
void bt(int l,int r)
{
    trlen++;
    int now;now=trlen;
    tr[now].l=l;tr[now].r=r;
    tr[now].lc=tr[now].rc=-1;
    tr[now].c=tr[now].jf=0;tr[now].cf=1;
    if(l==r)tr[now].c=a[l];
    else
    {
        int mid=(l+r)/2;
        tr[now].lc=trlen+1;bt(l,mid);
        tr[now].rc=trlen+1;bt(mid+1,r);
        tr[now].c=(tr[tr[now].lc].c+tr[tr[now].rc].c)%mod;
    }
}
void update(int now)
{
    int lc=tr[now].lc,rc=tr[now].rc;
    if(tr[now].cf!=1)
    {
        tr[lc].c*=tr[now].cf;tr[lc].c%=mod;
        tr[lc].cf*=tr[now].cf;tr[lc].cf%=mod;
        tr[lc].jf*=tr[now].cf;tr[lc].jf%=mod;
        tr[rc].c*=tr[now].cf;tr[rc].c%=mod;
        tr[rc].cf*=tr[now].cf;tr[rc].cf%=mod;
        tr[rc].jf*=tr[now].cf;tr[rc].jf%=mod;
    }
    if(tr[now].jf!=0)
    {
        tr[lc].c+=(tr[lc].r-tr[lc].l+1)*tr[now].jf;tr[lc].c%=mod;
        tr[lc].jf+=tr[now].jf;tr[lc].jf%=mod;
        tr[rc].c+=(tr[rc].r-tr[rc].l+1)*tr[now].jf;tr[rc].c%=mod;
        tr[rc].jf+=tr[now].jf;tr[rc].jf%=mod;
    }
    tr[now].cf=1,tr[now].jf=0;
}
void change1(int now,int l,int r,long long k)
{
    if(tr[now].l==l && tr[now].r==r)
    {
        tr[now].c*=k;tr[now].c%=mod;
        tr[now].cf*=k;tr[now].cf%=mod;
        tr[now].jf*=k;tr[now].jf%=mod;
        return ;
    }
    update(now);
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(r<=mid)change1(lc,l,r,k);
    else if(mid+1<=l)change1(rc,l,r,k);
    else change1(lc,l,mid,k),change1(rc,mid+1,r,k);
    tr[now].c=(tr[lc].c+tr[rc].c)%mod;
}
void change2(int now,int l,int r,long long k)
{
    if(tr[now].l==l && tr[now].r==r)
    {
        tr[now].c+=(tr[now].r-tr[now].l+1)*k;tr[now].c%=mod;
        tr[now].jf+=k;tr[now].jf%=mod;
        return ;
    }
    update(now);
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(r<=mid)change2(lc,l,r,k);
    else if(mid+1<=l)change2(rc,l,r,k);
    else change2(lc,l,mid,k),change2(rc,mid+1,r,k);
    tr[now].c=(tr[lc].c+tr[rc].c)%mod;
}
long long getsum(int now,int l,int r)
{
    if(tr[now].l==l && tr[now].r==r)return tr[now].c;
    update(now);
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(r<=mid)return getsum(lc,l,r);
    else if(mid+1<=l)return getsum(rc,l,r);
    else return (getsum(lc,l,mid)+getsum(rc,mid+1,r))%mod;
}
int main()
{
    scanf("%d%lld",&n,&mod);
    for(int i=1;i<=n;i++){scanf("%lld",&a[i]),a[i]%=mod;}
    trlen=0;bt(1,n);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int t,x,y;long long c;
        scanf("%d%d%d",&t,&x,&y);
        if(t==1)
        {
            scanf("%lld",&c);c%=mod;
            change1(1,x,y,c);
        }
        if(t==2)
        {
            scanf("%lld",&c);c%=mod;
            change2(1,x,y,c);
        }
        if(t==3)printf("%lld\n",getsum(1,x,y));
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值