HDU4578-Transformation-线段树的加、乘、变、次方操作

Sample Input

5 5
3 3 5 7
1 2 4 4
4 1 5 2
2 2 5 8
4 3 5 3
0 0

Sample Output

307
7489

题意：
给出n，m，表示该数有n个节点，m次操作。
接下来m次操作，op、x、y、z，
op=1时，将区间内每个元素+z
op=2时，将区间内每个元素*z
op=3时，将区间内每个元素都变成z
op=4时，求出区间内每个元素的p次方之和输出且对mod10007取余。

数组说明：

a[i]:该区间相等元素值
book[i]: 假设该节点下面的儿子节点全部相等则标记为0，否则为1

对于op=4时的次方操作：

if(L<=l&&r<=R&&book[i]==)
{
    ll ans=;
    for(int k=; k<p; k++)
        ans=(ans*a[i])%mod;
    ans=(ans*(r-l+))%mod;
    return ans;
}

 #include<stdio.h>
 #include<map>
 #include<iostream>
 #include<string.h>
 #include<algorithm>
 #include<stack>
 #include<cmath>
 #include<vector>
 using namespace std;
 typedef long long ll;
 const int mod=;
 const int N=1e5+;

 int a[*N];
 bool book[*N];

 void pushdown(int i)
 {
     if(book[i<<]||book[i<<|])
         book[i]=;
     else if(a[i<<]!=a[i<<|])
         book[i]=;
     else
     {
         book[i]=;
         a[i]=a[i<<]=a[i<<|];//因为a[i<<1]==a[i<<1|1]
     }
 }
 //1 x y 1 n
 void update(int i,int L,int R,int l,int r,int z,int op)
 {
     if(L<=l&&r<=R&&book[i]==)//1不相等 0相等
     {
         if(op==)
             a[i]=(a[i]+z)%mod;
         else if(op==)
             a[i]=(a[i]*z)%mod;
         else if(op==)
             a[i]=z;
         return;
     }
     if(book[i]==)
     {
         book[i<<]=book[i<<|]=;
         a[i<<]=a[i<<|]=a[i];
         book[i]=;
     }
     int mid=(l+r)>>;
     if(L<=mid)
         update(i<<,L,R,l,mid,z,op);
     if(R>mid)
         update(i<<|,L,R,mid+,r,z,op);
     pushdown(i);//a[i]=a[i<<1]+a[i<<1|1];
 }

 ll query(int i,int L,int R,int l,int r,int p)
 {
     if(L<=l&&r<=R&&book[i]==)
     {
         ll ans=;
         for(int k=;k<p;k++)
             ans=(ans*a[i])%mod;
         ans=(ans*(r-l+))%mod;
         return ans;
     }
     if(book[i]==)
     {
         book[i<<]=book[i<<|]=;
         a[i<<]=a[i<<|]=a[i];
         book[i]=;
     }
     int mid=(l+r)>>;
     ll ans=;
     if(L<=mid)
         ans+=query(i<<,L,R,l,mid,p);
     if(R>mid)
         ans+=query(i<<|,L,R,mid+,r,p);
     return ans%mod;

 }

 int main()
 {
     int n,m,op,L,R,z;
     while(cin>>n>>m)
     {
         if(n==&&m==)
             break;
         memset(book,,sizeof(book));//假设全相等标记为0//book[1]表示该节点下面的元素不相等
         memset(a,,sizeof(a));//该区间相等元素值
         for(int i=; i<m; i++)
         {
             cin>>op>>L>>R>>z;
             if(op!=) //1+,2*,3变为z,
                 update(,L,R,,n,z,op);
             else //*z次方
             {
                 ll ans=query(,L,R,,n,z)%mod;
                 printf("%lld\n",ans);
             }
         }
     }
     return ;
 }