Transformation(线段树)

题目链接:Transformation

大致题意

给定区间[1, n] 有如下四种操作:

​ ①将[l, r]区间的数值都增加c
​ ②将[l, r]区间的数值变为原来的c倍
​ ③将[l, r]区间的数值都变为c
​ ④查询[l, r]中每个点的p次方区间和, p = 1, 2, 3.

解题思路

这个题真的是太恶心了

本题考查线段树的 区间修改和区间查询

这个题由于每次查询是查单个点的p次方之和, 所以我们需要维护三个区间和, 分别表示p=1,2,3时的结果.

那么对于①②③的修改操作, 其中②③的操作都比较好处理和理解. 而对于操作①, 则需要进行公式推导. 推导过程请大家看别的大佬们的方式吧 这里就不再细说了~~(数学菜鸡留下了卑微的泪水)~~

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 1E5 + 10, mod = 10007;
struct node {
    int l, r;
    int sum1, sum2, sum3;
    int add, mul, lazy; //lazy表示变成一个数值的懒标签
}t[N << 2];

void pushdown(node& op, int add, int mul, int lazy) { //需要分类pushdown
    const int len = op.r - op.l + 1;
    if (lazy) {
        lazy %= mod;
        op.sum1 = lazy * len % mod;
        op.sum2 = lazy * lazy % mod * len % mod;
        op.sum3 = lazy * lazy % mod * lazy % mod * len % mod;
        
        op.add = 0, op.mul = 1; op.lazy = lazy;
    }
    
    if (mul != 1) { //对于乘法和加法的标签处理, 需要遵循先乘后加的原则
        op.sum1 = op.sum1 * mul % mod;
        op.sum2 = op.sum2 * mul % mod * mul % mod;
        op.sum3 = op.sum3 * mul % mod * mul % mod * mul % mod;
        
        op.add = mul * op.add % mod;
        op.mul = op.mul * mul % mod;
    }

    if (add) {
        op.sum3 = (op.sum3 + len * add % mod * add % mod * add % mod + 3 * op.sum2 % mod * add % mod + 3 * op.sum1 % mod * add % mod * add % mod) % mod;
        op.sum2 = (op.sum2 + 2 * add % mod * op.sum1 % mod + len * add % mod * add % mod) % mod;
        op.sum1 = (op.sum1 + len * add % mod) % mod;
        
        op.add = (op.add + add) % mod;
    }
}
void pushdown(int x) {
    node& p = t[x];
    pushdown(t[x << 1], p.add, p.mul, p.lazy);
    pushdown(t[x << 1 | 1], p.add, p.mul, p.lazy);
    p.add = 0, p.mul = 1, p.lazy = 0;
}

void pushup(node& p, node& l, node& r) {
    p.sum1 = (l.sum1 + r.sum1) % mod;
    p.sum2 = (l.sum2 + r.sum2) % mod;
    p.sum3 = (l.sum3 + r.sum3) % mod;
}
void pushup(int x) { pushup(t[x], t[x << 1], t[x << 1 | 1]); }

void build(int l, int r, int x = 1) {
    t[x] = { l, r, 0, 0, 0, 0, 1, 0 };
    if (l == r) return;
    int mid = l + r >> 1;
    build(l, mid, x << 1), build(mid + 1, r, x << 1 | 1);
    pushup(x);
}

void modify(int l, int r, int k, int c, int x = 1) { //k的值分别对应操作①②③
    if (l <= t[x].l && r >= t[x].r) { 
        if (k == 1) pushdown(t[x], c, 1, 0);
        else if (k == 2) pushdown(t[x], 0, c, 0);
        else pushdown(t[x], 0, 1, c);
        return;
    }
    pushdown(x);
    int mid = t[x].l + t[x].r >> 1;
    if (l <= mid) modify(l, r, k, c, x << 1);
    if (r > mid) modify(l, r, k, c, x << 1 | 1);
    pushup(x);
}

int ask(int l, int r, int k, int x = 1) {
    if (l <= t[x].l && r >= t[x].r) {
        if (k == 1) return t[x].sum1;
        if (k == 2) return t[x].sum2;
        return t[x].sum3;
    }
    pushdown(x);
    int mid = t[x].l + t[x].r >> 1;
    int res = 0;
    if (l <= mid) res += ask(l, r, k, x << 1);
    if (r > mid) res += ask(l, r, k, x << 1 | 1);
    return res % mod;
}

int main()
{
    int n, m;
    while (scanf("%d %d", &n, &m), n | m) {
        build(1, n);
        rep(i, m) {
            int op, l, r, c; scanf("%d %d %d %d", &op, &l, &r, &c);
            if (op == 4) printf("%d\n", ask(l, r, c));
            else modify(l, r, op, c);
        }
    }
    return 0;
}

QAQ这个题真的是太恶心了, 没处理过这么复杂的pushdown, 做题时总想一起维护所有懒标签, 不分类. 然后… 就没有然后了… (也可能是太菜了, 不会维护)

kuangbin线段树专题点这里!!!

END