[BZOJ1798] [AHOI2009]维护序列 &&[BZOJ5039] 序列维护

题目描述

Description

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

Input

第一行两个整数N和P(1≤P≤1000000000)。第二行含有N个非负整数,从左到右依次为a1,a2,…,aN, (0≤ai≤1000000000,1≤i≤N)。第三行有一个整数M,表示操作总数。从第四行开始每行描述一个操作,输入的操作有以下三种形式: 操作1:“1 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai×c (1≤t≤g≤N,0≤c≤1000000000)。 操作2:“2 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai+c (1≤t≤g≤N,0≤c≤1000000000)。 操作3:“3 t g”(不含双引号)。询问所有满足t≤i≤g的ai的和模P的值 (1≤t≤g≤N)。 同一行相邻两数之间用一个空格隔开,每行开头和末尾没有多余空格。

Output

对每个操作3,按照它在输入中出现的顺序,依次输出一行一个整数表示询问结果。

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

题目分析:

一个裸的线段树双标记
假设sum是节点保存的和 面对mul,add双标记:

      (sum+add*(a[num].y-a[num].x+1))*mul+add*(a[num].y-a[num].x+1)                                        =    mul*sum+(add*mul+add)*(a[num].y-a[num].x+1)

这样 我们可以在更新mul标记时直接将add标记乘上要乘的值

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m;
long long mod;
long long c[100100];
struct your
{
    long long mul;
    long long add;
    long long sum;
    int x,y;
}a[400400];
void build(int dx,int dy,int num)
{
    a[num].x=dx,a[num].y=dy;
    a[num].add=0,a[num].mul=1;
    if(dx==dy)
    {
        a[num].sum=c[dx]%mod;
        return ;
    }
    int mid=(dx+dy)>>1;
    build(dx,mid,num<<1);
    build(mid+1,dy,num<<1|1);
    a[num].sum=(a[num<<1].sum+a[num<<1|1].sum)%mod;
}
void push_down(int num)
{
    a[num<<1].add=(a[num<<1].add*a[num].mul+a[num].add)%mod;
    a[num<<1|1].add=(a[num<<1|1].add*a[num].mul+a[num].add)%mod;
    a[num<<1].mul=(a[num<<1].mul*a[num].mul)%mod;
    a[num<<1|1].mul=(a[num<<1|1].mul*a[num].mul)%mod;
    a[num<<1].sum=(a[num<<1].sum*a[num].mul+a[num].add*(a[num<<1].y-a[num<<1].x+1))%mod;
    a[num<<1|1].sum=(a[num<<1|1].sum*a[num].mul+a[num].add*(a[num<<1|1].y-a[num<<1|1].x+1))%mod;
    a[num].add=0;
    a[num].mul=1;
    return ;
}
void update_mul(int dx,int dy,long long val,int num)
{
    if(a[num].x==dx&&a[num].y==dy)
    {
        a[num].mul=(a[num].mul*val)%mod;
        a[num].add=(a[num].add*val)%mod;
        a[num].sum=(a[num].sum*val)%mod;
        return ;
    }
    if(a[num].mul!=1||a[num].add!=0)
        push_down(num);
    int mid=(a[num].x+a[num].y)>>1;
    if(dx>mid) update_mul(dx,dy,val,num<<1|1);
    else if(dy<=mid) update_mul(dx,dy,val,num<<1);
    else
    {
        update_mul(dx,mid,val,num<<1);
        update_mul(mid+1,dy,val,num<<1|1);
    }
    a[num].sum=(a[num<<1].sum+a[num<<1|1].sum)%mod;
    return ;
}
void update_add(int dx,int dy,long long val,int num)
{
    if(a[num].x==dx&&a[num].y==dy)
    {
        a[num].add=(a[num].add+val)%mod;
        a[num].sum=(a[num].sum+val*(a[num].y-a[num].x+1))%mod;
        return ;
    }
    if(a[num].mul!=1||a[num].add!=0)
        push_down(num);
    int mid=(a[num].x+a[num].y)>>1;
    if(dx>mid) update_add(dx,dy,val,num<<1|1);
    else if(dy<=mid) update_add(dx,dy,val,num<<1);
    else
    {
        update_add(dx,mid,val,num<<1);
        update_add(mid+1,dy,val,num<<1|1);
    }
    a[num].sum=(a[num<<1].sum+a[num<<1|1].sum)%mod;
    return ;
}
long long ask(int dx,int dy,int num)
{
    if(a[num].x==dx&&a[num].y==dy)
        return a[num].sum%mod;
    if(a[num].mul!=1||a[num].add!=0)
        push_down(num);
    int mid=(a[num].x+a[num].y)>>1;
    if(dx>mid) return ask(dx,dy,num<<1|1)%mod;
    if(dy<=mid) return ask(dx,dy,num<<1)%mod;
    return (ask(dx,mid,num<<1)+ask(mid+1,dy,num<<1|1))%mod;
}
int main()
{
    scanf("%d%lld",&n,&mod);
    for(int i=1;i<=n;i++)
        scanf("%lld",&c[i]);
    build(1,n,1);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int tmp;
        scanf("%d",&tmp);
        if(tmp==1)
        {
            int x,y;
            long long c;
            scanf("%d%d%lld",&x,&y,&c);
            update_mul(x,y,c,1);
        }
        if(tmp==2)
        {
            int x,y;
            long long c;
            scanf("%d%d%lld",&x,&y,&c);
            update_add(x,y,c,1);
        }
        if(tmp==3)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            printf("%lld\n",ask(x,y,1)%mod);
        }
    }
}

 

发表评论

邮箱地址不会被公开。 必填项已用*标注