HDU4578Transformation线段树的加、乘、变、次方操作

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]==0)
{
    ll ans=1;
    for(int k=0; k<p; k++)
        ans=(ans*a[i])%mod;
    ans=(ans*(r-l+1))%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=10007;
const int N=1e5+20;

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

void pushdown(int i)
{
    if(book[i<<1]||book[i<<1|1])
        book[i]=1;
    else if(a[i<<1]!=a[i<<1|1])
        book[i]=1;
    else
    {
        book[i]=0;
        a[i]=a[i<<1]=a[i<<1|1];//因为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]==0)//1不相等 0相等
    {
        if(op==1)
            a[i]=(a[i]+z)%mod;
        else if(op==2)
            a[i]=(a[i]*z)%mod;
        else if(op==3)
            a[i]=z;
        return;
    }
    if(book[i]==0)
    {
        book[i<<1]=book[i<<1|1]=0;
        a[i<<1]=a[i<<1|1]=a[i];
        book[i]=1;
    }
    int mid=(l+r)>>1;
    if(L<=mid)
        update(i<<1,L,R,l,mid,z,op);
    if(R>mid)
        update(i<<1|1,L,R,mid+1,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]==0)
    {
        ll ans=1;
        for(int k=0;k<p;k++)
            ans=(ans*a[i])%mod;
        ans=(ans*(r-l+1))%mod;
        return ans;
    }
    if(book[i]==0)
    {
        book[i<<1]=book[i<<1|1]=0;
        a[i<<1]=a[i<<1|1]=a[i];
        book[i]=1;
    }
    int mid=(l+r)>>1;
    ll ans=0;
    if(L<=mid)
        ans+=query(i<<1,L,R,l,mid,p);
    if(R>mid)
        ans+=query(i<<1|1,L,R,mid+1,r,p);
    return ans%mod;

}

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