【HDU4578 Transformation】线段树

题目链接：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次方之和。

思路：比较复杂的线段树操作。只有一个询问操作，那就是询问[l，r]之间数的p次方之和，我们不可能全部查询的节点，会TLE，最好的情况就是查询到一段[a，b]，这段区间内所有的值都相等，那么返回（b-a+1）*val 的值即可。 根据询问操作我们即可意识到我们要维护的是区间内所有值都相同的情况的区间。对于覆盖和加乘操作，我开始以为向下传递两层即可，这样是错误的，因为即可传递下去两层，还是有可能同时存在加乘操作。所以我们可以分两种情况讨论：1、当为覆盖操作时，直接覆盖区间即可，并且把标记的加乘操作赋为初始值。2、当为加乘操作时，先判断当前区间段是否为相同的值，是的话直接加乘，维护这个相同的值即可。如果不相同，看区间是否已有加乘标记，把这个加乘标记一直传递下去，知道遇见那个区间段区间所有值的相同停止。最后把这个加乘赋给最开始的区间段。简单的说就是，覆盖操作可直接维护，不是覆盖操作的话，区间只能保留一个加乘操作。

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 using namespace std;

 #define lz 2*u,l,mid
 #define rz 2*u+1,mid+1,r
 typedef long long lld;
 const int maxn=;
 const int mod=;
 lld add[*maxn], mul[*maxn], ch[*maxn], sum[*maxn];

 void build(int u, int l, int r)
 {
     mul[u]=;
     add[u]=sum[u]=;
     ch[u]=;
     if(l==r)
     {
         ch[u]=;
         return ;
     }
     int mid=(l+r)>>;
     build(lz);
     build(rz);
 }

 void push_down(int u, int l, int r)
 {
     if(l==r) return ;
     int mid=(l+r)>>;
     if(ch[u])
     {
         add[*u]=, mul[*u]=;
         add[*u+]=, mul[*u+]=;
         ch[*u]=ch[*u+]=;
         sum[*u]=sum[*u+]=sum[u];
         ch[u]=;
     }
     else
     {
         if(add[u])
         {
             if(ch[*u]) sum[*u]=(sum[*u]+add[u])%mod;
             else
             {
                 push_down(lz);
                 add[*u]=(add[*u]+add[u])%mod;
             }
             if(ch[*u+]) sum[*u+]=(sum[*u+]+add[u])%mod;
             else
             {
                 push_down(rz);
                 add[*u+]=(add[*u+]+add[u])%mod;
             }
             add[u]=;
         }
         if(mul[u]>)
         {
             if(ch[*u]) sum[*u]=(sum[*u]*mul[u])%mod;
             else
             {
                 push_down(lz);
                 mul[*u]=(mul[*u]*mul[u])%mod;
             }
             if(ch[*u+]) sum[*u+]=(sum[*u+]*mul[u])%mod;
             else
             {
                 push_down(rz);
                 mul[*u+]=(mul[*u+]*mul[u])%mod;
             }
             mul[u]=;
         }
     }
 }

 void Update(int u, int l, int r, int tl, int tr, int c, int op)
 {
     if(tl<=l&&r<=tr)
     {
         if(op==)
         {
             ch[u]=;
             mul[u]=, add[u]=;
             sum[u]=c;
         }
         else
         {
             if(ch[u])
             {
                 if(op==) sum[u]=(sum[u]+c)%mod;
                 else   sum[u]=sum[u]*c%mod;
             }
             else
             {
                 push_down(u,l,r);
                 if(op==) add[u]=(add[u]+c)%mod;
                 else mul[u]=mul[u]*c%mod;
             }
         }
         return ;
     }
     push_down(u,l,r);
     int mid=(l+r)>>;
     if(tr<=mid) Update(lz,tl,tr,c,op);
     else if(tl>mid) Update(rz,tl,tr,c,op);
     else
     {
         Update(lz,tl,mid,c,op);
         Update(rz,mid+,tr,c,op);
     }
 }

 lld Query(int u, int l, int r, int tl, int tr, int p)
 {
     if(tl<=l&&r<=tr)
     {
         if(ch[u])
         {
             lld ans=, tp=sum[u];
             for(int i=; i<=p; i++) ans=ans*tp%mod;
             return (r-l+)*ans%mod;
         }
     }
     push_down(u,l,r);
     int mid=(l+r)>>;
     if(tr<=mid) return Query(lz,tl,tr,p);
     else if(tl>mid) return Query(rz,tl,tr,p);
     else
     {
         lld t1=Query(lz,tl,mid,p);
         lld t2=Query(rz,mid+,tr,p);
         return (t1+t2)%mod;
     }
 }

 int main()
 {
     int n, m;
     while(cin >> n >> m, n+m)
     {
         build(,,n);
         for(int i=; i<=m; i++)
         {
             int l, r, c, op;
             scanf("%d%d%d%d",&op,&l,&r,&c);
             if(op<=) Update(,,n,l,r,c,op);
             else printf("%I64d\n",Query(,,n,l,r,c)%mod);
         }
     }
     return ;
 }