HDU 4578 Transformation --线段树，好题

题意： 给一个序列，初始全为0，然后有4种操作：

1. 给区间[L,R]所有值+c

2.给区间[L,R]所有值乘c

3.设置区间[L,R]所有值为c

4.查询[L,R]的p次方和（1<=p<=3）

解法： 线段树，维护三个标记,addmark,mulmark,setmark分别表示3种更新，然后p[1],p[2],p[3]分别表示该节点的1,2,3次方和。标记传递顺序setmark一定是第一个，因为setmark可以使mulmark,addmark都失效还原，mulmark和addmark的顺序倒是无所谓。

这题写了半个比赛时间，写的很迷，还是没写出来， 后来看了别人写的，才知道自己很多地方都没更新好甚至没更新。这题算是一道线段树的综合题了，混合了几种常见的更新，对线段树的整体把握很有帮助。

比如：

1.如果setmark有值，那么addmark,mulmark全部要还原。

2.如果mulmark有值，那么addmark也要更新，更新为addmark*mulmark

就是这两点一直没考虑到，WA了好久。

在addmark下传过程中更新p[1],p[2],p[3]的方法如下：

比如 a^3 -> (a+c)^3 的过程：  (a+c)^3 = a^3 + c^3 + 3a*c^2 + 3*a^2*c, a是变量， 所以提取出c，那么p[3]可以由p[1],p[2]推出，p[2]可以由p[1]推出。

即 p[3] = p[3] + c^3 + 3*c^2*p[1] + 3*c*p[2] .

p[2]同理可以由p[1]推出。

然后每次都做一下pushdown，就可以得出正确答案。

当时像优化一下，少做几次pushdown，即这次的操作类型与上次一样就不pushdown，结果那样就会出现问题，还不如每次都pushdown呢。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#define Mod 10007
#define SMod 10007
using namespace std;
#define N 100017

struct node{
    int p[];
    int setmark,addmark,mulmark;
}tree[*N];

void pushup(int rt){
    for(int i=;i<=;i++) tree[rt].p[i] = (tree[*rt].p[i] + tree[*rt+].p[i])%SMod;
}

void pushdown(int l,int r,int rt)
{
    int mid = (l+r)/;
    if(tree[rt].setmark)
    {
        int mk = tree[rt].setmark;
        tree[rt<<].setmark = tree[rt<<|].setmark = mk;
        tree[rt<<].addmark = tree[rt<<|].addmark = ;
        tree[rt<<].mulmark = tree[rt<<|].mulmark = ;

        tree[rt<<].p[] = (mid-l+)%Mod*mk%SMod;
        tree[rt<<|].p[] = (r-mid)%Mod*mk%SMod;

        tree[rt<<].p[] = (mid-l+)%Mod*mk%SMod*mk%SMod;
        tree[rt<<|].p[] = (r-mid)%Mod*mk%SMod*mk%SMod;

        tree[rt<<].p[] = (mid-l+)%Mod*mk%SMod*mk%SMod*mk%SMod;
        tree[rt<<|].p[] = (r-mid)%Mod*mk%SMod*mk%SMod*mk%SMod;
        tree[rt].setmark = ;
    }
    if(tree[rt].mulmark != )
    {
        int mk = tree[rt].mulmark;
        tree[rt<<].mulmark *= mk, tree[rt<<].mulmark%=SMod;
        tree[rt<<|].mulmark *= mk, tree[rt<<|].mulmark%=SMod;

        tree[rt<<].addmark = tree[rt<<].addmark%SMod*mk%SMod;
        tree[rt<<|].addmark = tree[rt<<|].addmark%SMod*mk%SMod;

        tree[rt<<].p[] = tree[rt<<].p[]%Mod*mk%SMod;
        tree[rt<<|].p[] = tree[rt<<|].p[]%Mod*mk%SMod;

        tree[rt<<].p[] = tree[rt<<].p[]%Mod*mk%SMod*mk%SMod;
        tree[rt<<|].p[] = tree[rt<<|].p[]%Mod*mk%SMod*mk%SMod;

        tree[rt<<].p[] = tree[rt<<].p[]%Mod*mk%SMod*mk%SMod*mk%SMod;
        tree[rt<<|].p[] = tree[rt<<|].p[]%Mod*mk%SMod*mk%SMod*mk%SMod;
        tree[rt].mulmark = ;
    }
    if(tree[rt].addmark)
    {
        int mk = tree[rt].addmark;
        tree[rt<<].addmark += mk, tree[rt<<].addmark%SMod;
        tree[rt<<|].addmark += mk, tree[rt<<|].addmark%SMod;

        tree[rt<<].p[] = (tree[rt<<].p[]%Mod + (mid-l+)%Mod*mk%SMod*mk%SMod*mk%SMod + *mk%Mod*tree[rt<<].p[]%SMod + *mk%SMod*mk%SMod*tree[rt<<].p[]%SMod)%SMod;
        tree[rt<<|].p[] = (tree[rt<<|].p[]%Mod + (r-mid)%Mod*mk%SMod*mk%SMod*mk%SMod + *mk%Mod*tree[rt<<|].p[]%SMod + *mk%SMod*mk%SMod*tree[rt<<|].p[]%SMod)%SMod;

        tree[rt<<].p[] = (tree[rt<<].p[]%Mod + (mid-l+)%Mod*mk%SMod*mk%SMod + *mk%Mod*tree[rt<<].p[]%SMod)%SMod;
        tree[rt<<|].p[] = (tree[rt<<|].p[]%Mod + (r-mid)%Mod*mk%SMod*mk%SMod + *mk%Mod*tree[rt<<|].p[]%SMod)%SMod;

        tree[rt<<].p[] = (tree[rt<<].p[]%Mod + (mid-l+)%Mod*mk%SMod)%SMod;
        tree[rt<<|].p[] = (tree[rt<<|].p[]%Mod + (r-mid)%Mod*mk%SMod)%SMod;
        tree[rt].addmark = ;
    }
}

void build(int l,int r,int rt)
{
    tree[rt].setmark = tree[rt].addmark = ;
    tree[rt].mulmark = ;
    memset(tree[rt].p,,sizeof(tree[rt].p));
    if(l == r) return;
    int mid = (l+r)/;
    build(l,mid,rt<<);
    build(mid+,r,rt<<|);
}

void update(int l,int r,int aa,int bb,int tag,int val,int rt)  //tag = 3 : set  2:mul 1: add
{
    if(aa <= l && bb >= r)
    {
        if(tag == )
        {
            tree[rt].p[] = (r-l+)%Mod*val%SMod;
            tree[rt].p[] = (r-l+)%Mod*val%SMod*val%SMod;
            tree[rt].p[] = (r-l+)%Mod*val%SMod*val%SMod*val%SMod;
            tree[rt].setmark = val;
            tree[rt].addmark = ;
            tree[rt].mulmark = ;
        }
        else if(tag == )
        {
            tree[rt].p[] = tree[rt].p[]%SMod*val%SMod;
            tree[rt].p[] = tree[rt].p[]%SMod*val%SMod*val%SMod;
            tree[rt].p[] = tree[rt].p[]%SMod*val%SMod*val%SMod*val%SMod;
            tree[rt].mulmark = tree[rt].mulmark%SMod*val%SMod;
            tree[rt].addmark = tree[rt].addmark%SMod*val%SMod;
        }
        else if(tag == )
        {
            tree[rt].p[] = (tree[rt].p[]%SMod + (r-l+)%SMod*val%SMod*val%SMod*val%SMod + *val%SMod*tree[rt].p[]%SMod + *val%SMod*val%SMod*tree[rt].p[]%SMod)%SMod;
            tree[rt].p[] = (tree[rt].p[]%SMod + (r-l+)%SMod*val%SMod*val%SMod + *val%SMod*tree[rt].p[]%SMod)%SMod;
            tree[rt].p[] = (tree[rt].p[]%SMod + (r-l+)%SMod*val%SMod)%SMod;
            tree[rt].addmark = (tree[rt].addmark+val)%SMod;
        }
        return;
    }
    int mid = (l+r)/;
    pushdown(l,r,rt);
    if(aa <= mid) update(l,mid,aa,bb,tag,val,rt<<);
    if(bb > mid)  update(mid+,r,aa,bb,tag,val,rt<<|);
    pushup(rt);
}

int query(int l,int r,int aa,int bb,int i,int rt)
{
    if(aa > r || bb < l) return ;
    if(aa <= l && bb >= r)
        return tree[rt].p[i];
    pushdown(l,r,rt);
    int res = ;
    int mid = (l+r)/;
    return (query(l,mid,aa,bb,i,rt<<)%SMod+query(mid+,r,aa,bb,i,rt<<|)%SMod)%SMod;
}

int main()
{
    int n,m,i,j,op,x,y,c;
    while(scanf("%d%d",&n,&m)!=EOF && n+m)
    {
        build(,n,);
        while(m--)
        {
            scanf("%d%d%d%d",&op,&x,&y,&c);
            if(op != ) update(,n,x,y,op,c,);
            else        printf("%d\n",query(,n,x,y,c,)%SMod);
        }
    }
    return ;
}