HDU-4578-Transformation

这个题感觉好麻烦，可能是以前没有做过类似复杂的线段树的原因。这个题有4种操作，分别为

1、对区间[l,r]加上一个值.

2、对区间[l,r]乘以一个值

3、对区间[l,r]全部复制为x;

4、询问区间[l,r]的p次方和

这个题由于p比较小，所以可以用3个变量或大小为3的数组存放值。然后结点里存的分别是当前区间增加操作值、乘操作值以及是否全部赋值为同一值。对于每次更新的时候需要把先乘再加，即把当前的大区间乘的信息传到子区间去。然后再更新加的操作值。另外，还有需要注意的就是更新的时候可以写成展开式的形式,eg (a+b)^2=a^2+2*a*b+b^2，具体看代码吧

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int mod=10007;
const int maxn=1e5+10;
const int maxm=maxn*4;
struct Node
{
    int l;
    int r;
    int val;
    int add;
    int mul;
    bool issame;
    int ans[3];
} t[maxm];
int n,m;
void Build(int l,int r,int index)
{
    t[index].l=l;
    t[index].r=r;
    t[index].val=0;
    t[index].add=0;
    t[index].mul=1;
    memset(t[index].ans,0,sizeof(t[index].ans));
    if(l==r)
        return;
    int mid=(l+r)>>1;
    Build(l,mid,index<<1);
    Build(mid+1,r,index<<1|1);
}
void Calc(int index,int mul,int add)
{
    int a=t[index].ans[0],a2=t[index].ans[1],a3=t[index].ans[2];
    t[index].ans[0]=(t[index].ans[0]*mul%mod+add*(t[index].r-t[index].l+1))%mod;
    t[index].ans[1]=(t[index].ans[1]*mul%mod*mul%mod+2*a*add%mod*mul%mod+add*add%mod*(t[index].r-t[index].l+1)%mod)%mod;
    t[index].ans[2]=(t[index].ans[2]*mul%mod*mul%mod*mul%mod+3*a2*add%mod*mul%mod*mul%mod+3*a*add%mod*add%mod*mul%mod+add*add%mod*add%mod*(t[index].r-t[index].l+1)%mod)%mod;
}
void PushUp(int index)
{
    t[index].ans[0]=(t[index<<1].ans[0]+t[index<<1|1].ans[0])%mod;
    t[index].ans[1]=(t[index<<1].ans[1]+t[index<<1|1].ans[1])%mod;
    t[index].ans[2]=(t[index<<1].ans[2]+t[index<<1|1].ans[2])%mod;
}
void PushDown(int index)
{
    if(t[index].l==t[index].r)
        return;
    if(!t[index].issame&&!t[index].add&&t[index].mul==1)
        return;
    if(t[index].issame)
    {
        memset(t[index<<1].ans,0,sizeof(t[index<<1].ans));
        memset(t[index<<1|1].ans,0,sizeof(t[index<<1|1].ans));
        t[index<<1].add=0;
        t[index<<1].mul=1;
        t[index<<1].ans[0]=t[index].val*(t[index<<1].r-t[index<<1].l+1)%mod;
        t[index<<1].ans[1]=t[index].val*t[index].val%mod*(t[index<<1].r-t[index<<1].l+1)%mod;
        t[index<<1].ans[2]=t[index].val*t[index].val%mod*t[index].val%mod*(t[index<<1].r-t[index<<1].l+1)%mod;
        t[index<<1|1].add=0;
        t[index<<1|1].mul=1;
        t[index<<1|1].ans[0]=t[index].val*(t[index<<1|1].r-t[index<<1|1].l+1)%mod;
        t[index<<1|1].ans[1]=t[index].val*t[index].val%mod*(t[index<<1|1].r-t[index<<1|1].l+1)%mod;
        t[index<<1|1].ans[2]=t[index].val*t[index].val%mod*t[index].val%mod*(t[index<<1|1].r-t[index<<1|1].l+1)%mod;
        t[index<<1].val=t[index<<1|1].val=t[index].val;
        t[index<<1].issame=t[index<<1|1].issame=true;
        t[index].issame=false;
    }
    if(t[index].add||t[index].mul!=1)
    {
        t[index<<1].add=(t[index<<1].add*t[index].mul%mod+t[index].add)%mod;
        t[index<<1].mul=t[index<<1].mul*t[index].mul%mod;
        Calc(index<<1,t[index].mul,t[index].add);
        t[index<<1|1].add=(t[index<<1|1].add*t[index].mul%mod+t[index].add)%mod;
        t[index<<1|1].mul=(t[index<<1|1].mul*t[index].mul)%mod;
        Calc(index<<1|1,t[index].mul,t[index].add);
    }
    t[index].mul=1;
    t[index].add=0;
}
void Update(int l,int r,int index,int val,int op)
{
    PushDown(index);
    if(t[index].l==l&&t[index].r==r)
    {
        if(op==1)
        {
            t[index].add=(t[index].add+val)%mod;
            Calc(index,1,val);
        }
        else if(op==2)
        {
            t[index].mul=(t[index].mul*val)%mod;
            Calc(index,val,0);
        }
        else
        {
            t[index].mul=1;
            t[index].add=0;
            t[index].issame=true;
            t[index].val=val;
            memset(t[index].ans,0,sizeof(t[index].ans));
            Calc(index,1,val);
        }
        return;
    }
    int mid=(t[index].l+t[index].r)>>1;
    if(r<=mid)
        Update(l,r,index<<1,val,op);
    else if(l>mid)
        Update(l,r,index<<1|1,val,op);
    else
    {
        Update(l,mid,index<<1,val,op);
        Update(mid+1,r,index<<1|1,val,op);
    }
    PushUp(index);
}
int Query(int l,int r,int index,int p)
{
    PushDown(index);
    if(t[index].l==l&&t[index].r==r)
        return t[index].ans[p-1];
    int mid=(t[index].l+t[index].r)>>1;
    if(r<=mid)
        return Query(l,r,index<<1,p);
    else if(l>mid)
        return Query(l,r,index<<1|1,p);
    else
        return (Query(l,mid,index<<1,p)+Query(mid+1,r,index<<1|1,p))%mod;
}
int main()
{
    while(scanf("%d%d",&n,&m)&&(n+m))
    {
        Build(1,n,1);
        for(int i=0; i<m; i++)
        {
            int op,x,y,v;
            scanf("%d%d%d%d",&op,&x,&y,&v);
            if(op==1)
                Update(x,y,1,v,1);
            else if(op==2)
            {
                if(v<=1)
                    continue;
                Update(x,y,1,v,2);
            }
            else if(op==3)
                Update(x,y,1,v,3);
            else
                printf("%d\n",Query(x,y,1,v));
        }
    }
    return 0;
}