HDU 4578(线段树)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4578

 

题目大意：有四个操作，1、对l到r数字全加c 2、对l到r数字全乘以c 3、对l到r数字全变成c 4.查询l到r中数字p次方和

 

题目思路：操作相当多，但是我们可以把他简化成两个：加和乘。对于一个乘可以理解成乘标记*这个数字，同时加标记*这个数字。如果只有一个加，那就直接加标记+这个数字。所以对于加，乘标记为1，加标记为c。对于乘，乘标记为c，加标记为0。对于置c，乘标记为0，加标记为c。由此我们就确定了更新延迟标记以及更新区间的方法。对于查询p次方，平方和立方需要展开。

展开方式如下：

图片来自：https://www.cnblogs.com/Cwolf9/p/9513273.html

剩下的看代码基本没问题了。

 

以下是代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
const int MAXN = 1e5+5;
const int MOD = 10007;
struct node{
    int l,r;
    ll sum1,sum2,sum3,mark1,mark2;
}a[MAXN<<2];
void build(int rt,int l,int r){
    a[rt].l=l,a[rt].r=r;
    a[rt].sum1=a[rt].sum2=a[rt].sum3=a[rt].mark2=0;
    a[rt].mark1=1;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
}
void cal(int rt,ll x,ll b){
    int len=a[rt].r-a[rt].l+1;
    a[rt].sum3=(((x*x*x)%MOD*a[rt].sum3)%MOD+((b*b*b)%MOD*len)%MOD+((3ll*x*b*b)%MOD*a[rt].sum1)%MOD+((3ll*x*x*b)%MOD*a[rt].sum2)%MOD)%MOD;
    a[rt].sum2=((x*x*a[rt].sum2)%MOD+(2ll*x*b*a[rt].sum1)%MOD+(b*b*len)%MOD)%MOD;
    a[rt].sum1=((x*a[rt].sum1)%MOD+b*len%MOD)%MOD;
    a[rt].mark1=(a[rt].mark1*x)%MOD;
    a[rt].mark2=(a[rt].mark2*x+b)%MOD;
}
void spread(int rt){
    if(a[rt].mark1!=1||a[rt].mark2){
        cal(rt<<1,a[rt].mark1,a[rt].mark2);
        cal(rt<<1|1,a[rt].mark1,a[rt].mark2);
        a[rt].mark1=1,a[rt].mark2=0;
    }
}
void update(int rt,int l,int r,ll x,ll b){
    if(a[rt].l>=l&&a[rt].r<=r){
        cal(rt,x,b);
        return;
    }
    spread(rt);
    int mid=(a[rt].l+a[rt].r)>>1;
    if(l<=mid)update(rt<<1,l,r,x,b);
    if(r>mid)update(rt<<1|1,l,r,x,b);
    a[rt].sum1=(a[rt<<1].sum1+a[rt<<1|1].sum1)%MOD;
    a[rt].sum2=(a[rt<<1].sum2+a[rt<<1|1].sum2)%MOD;
    a[rt].sum3=(a[rt<<1].sum3+a[rt<<1|1].sum3)%MOD;
}
ll query(int rt,int l,int r,int p){
    if(a[rt].l>=l&&a[rt].r<=r){
        if(p==1)return a[rt].sum1;
        else if(p==2)return a[rt].sum2;
        else return a[rt].sum3;
    }
    int mid=(a[rt].l+a[rt].r)>>1;
    spread(rt);
    ll ans=0;
    if(l<=mid)ans=(ans+query(rt<<1,l,r,p))%MOD;
    if(r>mid)ans=(ans+query(rt<<1|1,l,r,p))%MOD;
    return ans;
}
int main(){
    int n,m,p,x,y,z;
    while(~scanf("%d%d",&n,&m)&&n&&m){
        build(1,1,n);
        while(m--){
            scanf("%d%d%d%d",&p,&x,&y,&z);
            if(p==1){
                update(1,x,y,1,z);
            }
            else if(p==2){
                update(1,x,y,z,0);
            }
            else if(p==3){
                update(1,x,y,0,z);
            }
            else{
                ll ans=query(1,x,y,z)%MOD;
                printf("%lld\n",ans);
            }
        }
    }
    return 0;
}