HDU - 4578 Transformation 线段树 思维

题目链接： HDU - 4578 Transformation

题意：给你一个数组，初始值为零，有四种操作

（1）"1 x y c",代表 把区间 [x,y] 上的值全部加c

（2）"2 x y c",代表 把区间 [x,y] 上的值全部乘以c

（3）"3 x y c" 代表 把区间 [x,y]上的值全部赋值为c

（4）"4 x y p" 代表 输出区间 [x,y] 上值的p次方和1<=p<=3
思路：

一开始想要用4个标记数组分别标记，但第四种操作没想到该如何写 （太菜了），只好去借鉴大佬的博客，发现了一种新的方法

查询到一段[l，r]，这段区间内所有的值都相等，那么返回（r-l+1）*val 的值即可。只要判断区间内的所有数是否相同，并记录下区间内所有数相同的区间的值即可。每次询问时查询到区间内所有值相同的区间即可

 

参考博客：Wang_128

#include<cstdio>
#include<cstring>
#include<algorithm>
#define mod 10007
using namespace std;
const int maxn=262144;
int sum[maxn],vis[maxn];  //sum[rt] 代表此节点子树所在区间都为这个值
void merge(int rt)
{
    if( !vis[rt<<1] || !vis[rt<<1|1])
        vis[rt] = 0;
    else if( sum[rt<<1] != sum[rt<<1|1] )
        vis[rt] = 0;
    else
        vis[rt] = 1 , sum[rt] = sum[rt<<1];   // 只有当这个节点的左右子树的值相同，且标记为1时，他所代表的区间的值都相同
}
void update(int f,int L,int R,int c,int l,int r,int rt)
{
    if(L<=l && R>=r && vis[rt])
    {
        if(f == 1)
            sum[rt] = (sum[rt]+c)%mod;
        else if(f == 2)
            sum[rt] = sum[rt]*c%mod;
        else
            sum[rt] = c%mod;
        return ;
    }
    if(vis[rt])
    {
        sum[rt<<1] = sum[rt<<1|1] = sum[rt];
        vis[rt<<1] = vis[rt<<1|1] = 1;
        vis[rt] = 0;
    }
    //叶子结点的标记永远为 1
    int m = (l+r)>>1;
    if(L<=m)
        update(f,L,R,c,l,m,rt<<1);
    if(R>m)
        update(f,L,R,c,m+1,r,rt<<1|1);
    merge(rt);
}
int query(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l && R>=r && vis[rt])
    {
        int ans=1;
        for(int i=1;i<=c;i++)
            ans = ans*sum[rt]%mod;
        ans = ans*(r-l+1)%mod;
        return ans;
    }
    if(vis[rt])
    {
        sum[rt<<1] = sum[rt<<1|1] = sum[rt];
        vis[rt<<1] = vis[rt<<1|1] = 1;
        vis[rt] = 0;
    }
    int m=(l+r)>>1;
    int cnt=0;
    if(L<=m)
        cnt += query(L,R,c,l,m,rt<<1);
    if(R>m)
        cnt += query(L,R,c,m+1,r,rt<<1|1);
    return cnt%mod;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m) && (n||m))
    {
        memset(vis,1,sizeof(vis));
        memset(sum,0,sizeof(sum));
        int f,x,y,c;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d%d",&f,&x,&y,&c);
            if(f==4)
                printf("%d\n",query(x,y,c,1,n,1));
            else
                update(f,x,y,c,1,n,1);
        }
    }
    return 0;
}