【题解】hdu4578（2018-07-27校赛 线段树入门 C）复杂的线段树操作

题目

题目链接
记得比赛时没有做出来，应该是去学了某个大佬的题解，但是已经找不到原作者了。

#include<cstdio>
#define lc (o<<1)
#define rc (o<<1|1)
#define mid (l+r>>1)
const int N=1e5+5;
const int mod=10007;
typedef long long ll;
ll sum1[N<<2],sum2[N<<2],sum3[N<<2];
//记录和、平方和、立方和
ll add[N<<2],mul[N<<2],set[N<<2];
//增加、乘积、更改标记 
void push_up(int o)
{
    sum1[o]=(sum1[lc]+sum1[rc])%mod;
    sum2[o]=(sum2[lc]+sum2[rc])%mod;
    sum3[o]=(sum3[lc]+sum3[rc])%mod;
}
void push_down(int o,int l,int r)
{
    if(set[o]){
        set[lc]=set[rc]=set[o];
        add[lc]=add[rc]=0;
        mul[lc]=mul[rc]=1;
        ll tmp=((set[o]*set[o])%mod)*set[o]%mod;
        sum1[lc]=(((r-l+1)-((r-l+1)>>1))%mod)*(set[o]%mod)%mod;
        sum1[rc]=(((r-l+1)>>1)%mod)*(set[o]%mod)%mod;
        sum2[lc]=(((r-l+1)-((r-l+1)>>1))%mod)*((set[o]*set[o])%mod)%mod;
        sum2[rc]=(((r-l+1)>>1)%mod)*((set[o]*set[o])%mod)%mod;
        sum3[lc]=(((r-l+1)-((r-l+1)>>1))%mod)*tmp%mod;
        sum3[rc]=(((r-l+1)>>1)%mod)*tmp%mod;
        set[o]=0;
    }
    if(mul[o]!=1){   
        mul[lc]=(mul[lc]*mul[o])%mod;
        mul[rc]=(mul[rc]*mul[o])%mod;
        if(add[lc])
            add[lc]=(add[lc]*mul[o])%mod;
        if(add[rc])
            add[rc]=(add[rc]*mul[o])%mod;
        ll tmp=(((mul[o]*mul[o])%mod*mul[o])%mod);
        sum1[lc]=(sum1[lc]*mul[o])%mod;
        sum1[rc]=(sum1[rc]*mul[o])%mod;
        sum2[lc]=(sum2[lc]%mod)*((mul[o]*mul[o])%mod)%mod;
        sum2[rc]=(sum2[rc]%mod)*((mul[o]*mul[o])%mod)%mod;
        sum3[lc]=(sum3[lc]%mod)*tmp%mod;
        sum3[rc]=(sum3[rc]%mod)*tmp%mod;
        mul[o]=1;
    }
    if(add[o]){
        add[lc]+=add[o]; 
        add[rc]+=add[o];
        ll tmp=(add[o]*add[o]%mod)*add[o]%mod;
        sum3[lc]=(sum3[lc]+(tmp*((r-l+1)-((r-l+1) >> 1))%mod)+3*add[o]*((sum2[lc]+sum1[lc]*add[o])%mod))%mod;
        sum3[rc]=(sum3[rc]+(tmp*((r-l+1)>>1)%mod)+ 3 * add[o] * ((sum2[rc] + sum1[rc] * add[o]) % mod)) % mod;
        sum2[lc]=(sum2[lc]+((add[o]*add[o]%mod)*((r-l+1)-((r-l+1)>>1))%mod) + (2 * sum1[lc] * add[o] % mod)) % mod;
        sum2[rc]=(sum2[rc]+(((add[o]*add[o]%mod)*((r-l+1)>>1))%mod)+(2*sum1[rc]*add[o]%mod))%mod;
        sum1[lc]=(sum1[lc]+((r-l+1)-((r-l+1)>>1))*add[o])% mod;
        sum1[rc]=(sum1[rc]+((r-l+1)>>1)*add[o])%mod;
        add[o]=0;
    }
}
void update(int o,int l,int r,int nl,int nr,int op,int c)
{
    if(nl<=l&&nr>=r){
        if(op==3) {
            set[o]=c;
            add[o]=0;
            mul[o]=1;
            sum1[o]=((r-l+1)*c)%mod;
            sum2[o]=((r-l+1)*((c*c)%mod))%mod;
            sum3[o]=((r-l+1)*(((c*c)%mod)*c%mod))%mod;
        }
        else if(op==2){
            mul[o]=(mul[o]*c)%mod;
            if(add[o])
                add[o]=(add[o]*c)%mod;
            sum1[o]=(sum1[o]*c)%mod;
            sum2[o]=(sum2[o]*(c*c%mod))%mod;
            sum3[o]=(sum3[o]*((c*c%mod)*c%mod))%mod;
        } 
        else if(op==1) {
            add[o]+=c;
            ll tmp=(((c*c)%mod*c)%mod*(r-l+1))%mod;
            sum3[o]=(sum3[o]+tmp+3*c*((sum2[o]+sum1[o]*c)%mod))%mod;
            sum2[o]=(sum2[o]+(c*c%mod*(r-l+1)%mod)+2*sum1[o]*c)%mod;
            sum1[o]=(sum1[o]+(r-l+1)*c)%mod;
        }
        return;
    }
    push_down(o,l,r);
    if(nl>mid)
        update(rc,mid+1,r,nl,nr,op,c);
    else if(nr<=mid)
        update(lc,l,mid,nl,nr,op,c);
    else {
        update(lc,l,mid,nl,mid,op,c);
        update(rc,mid+1,r,mid+1,nr,op,c);
    }
    push_up(o);
}
void build(int o,int l,int r)
{
    add[o]=set[o]=0;
    mul[o]=1;
    if(l==r)
    {
        sum1[o]=sum2[o]=sum3[o]=0;
        return;
    }
    build(lc,l,mid);build(rc,mid+1,r);
    push_up(o);
}
ll query(int o,int l,int r,int nl,int nr,int p)
{
    if(nl<=l&&r<=nr)
    {
        if(p==1)
        return sum1[o]%mod;
        else if(p==2)
        return sum2[o]%mod;
        else return sum3[o]%mod;
    }
    push_down(o,l,r);
    if(nr<=mid)return query(lc,l,mid,nl,nr,p);
    else if(nl>mid)return query(rc,mid+1,r,nl,nr,p);
    else return (query(lc,l,mid,nl,mid,p)+query(rc,mid+1,r,mid+1,nr,p))%mod;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int n,m;
    while(scanf("%d%d",&n,&m)&&n&&m)
    {
        int a,b,c,op;
        build(1,1,n);
        while(m--)
        {
            scanf("%d %d %d %d",&op,&a,&b,&c);
            if(op!=4)update(1,1,n,a,b,op,c);
            else printf("%lld\n",query(1,1,n,a,b,c));
        }
    }
    return 0;
}

总结

没做出来的原因：操作数太多，其中的平方和立方操作还加上替换操作，感觉很难受。
没有想到通过对平方和立方和进行公式展开，然后建多棵线段树来进行维护的方法。
注意本题维护多棵线段树进行操作时的先后顺序。