HDU - 4578 Transformation(线段树 区间修改)

题目大意:给出N个数,初始化都是0,4种操作
1 x y c:区间[x,y]的所有数加上c
2 x y c:区间[x,y]的所有数乘上c
3 x y c:区间[x,y]的所有数变成c
4 x y c:求区间[x,y]的所有数的c次方和

解题思路:维护区间的4个数,val表示区间内相同的数,add表示区间内的所有数加上了多少,mul表示区间内的所有数乘上了多少,same表示区间内的所有数是否都相等
初始化的时候就是val是0,add是0,mul是1,same除了叶结点是1外,其他的都是0
如果区间内的数都相等的话,求和就很方便了
接下来讨论怎么修改
如果是操作3,找到区间,直接就修改val,然后将same和mul变成1,add变成0即可
如果是加或者乘操作,就要分情况讨论了
首先,所要修改的区间的same == 1,那么直接修改val就可以了
如果不是相同的话,就要先PushDown了,使得结果是最新的,然后再进行修改,修改的时候只需要修改相应的add或者mul就可以了

现在讨论一下PushDown,PushDown并不是单纯的只修改父结点和子结点就完了,如果这样修改的话,后面会出现加乘的交替的,到时就不知道哪个前哪个后了,所以要进行递归修改,将所有的标记都往下移,知道same == 1,如此,就不会出现加乘交替的情况了

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 100010 << 2;
const int MOD = 10007;
LL add[N], mul[N], same[N], val[N];
int n, m;

void build(int u, int l, int r) {
    add[u] = val[u] = same[u] = 0;
    mul[u] = 1;
    if (l == r) {
        same[u] = 1;
        return ;
    }
    int mid = (l + r) >> 1;
    build(u << 1, l, mid);
    build(u << 1 | 1, mid + 1, r);
}

void PushDown(int u, int l, int r) {
    if (l == r) return ;
    int mid = (l + r) >> 1;
    if (same[u]) {
        same[u << 1] = same[u << 1 | 1] = 1;
        add[u << 1] = add[u << 1 | 1] = 0;
        mul[u << 1] = mul[u << 1 | 1] = 1;
        val[u << 1] = val[u << 1 | 1] = val[u];
        same[u] = 0;
    }
    else {
        if (add[u]) {
            if (same[u << 1]) val[u << 1] = (val[u << 1] + add[u]) % MOD;
            else {
                PushDown(u << 1, l, mid);
                add[u << 1] = (add[u << 1] + add[u]) % MOD;
            }
            if (same[u << 1 | 1]) val[u << 1 | 1] = (val[u << 1 | 1] + add[u]) % MOD;
            else {
                PushDown(u << 1 | 1, mid + 1, r);
                add[u << 1 | 1] = (add[u << 1 | 1] + add[u]) % MOD;
            }
            add[u] = 0;
        }
        else if (mul[u] > 1) {
            if (same[u << 1]) val[u << 1] = (val[u << 1] * mul[u]) % MOD;
            else {
                PushDown(u << 1, l, mid);
                mul[u << 1] = (mul[u << 1] * mul[u]) % MOD;
            }
            if (same[u << 1 | 1]) val[u << 1 | 1] = (val[u << 1 | 1] * mul[u]) % MOD;
            else {
                PushDown(u << 1 | 1, mid + 1, r);
                mul[u << 1 | 1] = (mul[u << 1 | 1] * mul[u]) % MOD;
            }
            mul[u] = 1;
        }
    }
}

void Modify(int u, int l, int r, int L, int R, int c, int op) {
    if (L <= l && r <= R) {
        if (op == 3) {
            mul[u] = same[u] = 1;
            add[u] = 0;
            val[u] = c;
        }
        else {
            if (same[u]) {
                if (op == 1) val[u] = (val[u] + c) % MOD;
                else val[u] = (val[u] * c ) % MOD;
            }
            else {
                PushDown(u, l, r);
                if (op == 1) add[u] = (add[u] + c) % MOD;
                else mul[u] = (mul[u] * c) % MOD;
            }
        }
        return ;
    }
    int mid = (l + r) >> 1;
    PushDown(u, l, r);
    if (R <= mid) Modify(u << 1, l, mid, L, R, c, op);
    else if (L > mid) Modify(u << 1 | 1, mid + 1, r, L, R, c, op);
    else {
        Modify(u << 1, l, mid, L, mid, c, op);
        Modify(u << 1 | 1, mid + 1, r, mid + 1, R, c, op);
    }
}

LL Query(int u, int l, int r, int L, int R, int p) {
    if (L <= l && r <= R) {
        if (same[u]) {
            LL ans = 1, tmp = val[u];
            for (int i = 1; i <= p; i++) ans = (ans * tmp) % MOD;
            return ans * (r - l + 1) % MOD;
        }
    }
    PushDown(u, l, r);
    int mid = (l + r) >> 1;
    if (R <= mid) return Query(u << 1, l, mid, L, R, p);
    else if (L > mid) return Query(u << 1 | 1, mid + 1, r, L, R, p);
    else return (Query(u << 1, l, mid, L, mid, p) + Query(u << 1 | 1, mid + 1, r, mid + 1, R, p)) % MOD;
}

void solve() {
    build(1, 1, n);
    int op, a, b, c;
    while (m--) {
        scanf("%d%d%d%d", &op, &a, &b, &c);
        if (op < 4) Modify(1, 1, n, a, b, c, op);
        else printf("%lld\n", Query(1, 1, n, a, b, c));
    }
}

int main() {
    while (scanf("%d%d", &n, &m) != EOF && n + m) solve();
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值