【线段树】【AHOI 2009】维护序列

题目链接：传送门

 

---

 

题目描述

老师交给小可可一个维护数列的任务，现在小可可希望你来帮他完成。 有长为N的数列，不妨设为a1,a2,…,aN 。有如下三种操作形式：
(1)把数列中的一段数全部乘一个值;
(2)把数列中的一段数全部加一个值;
(3)询问数列中的一段数的和，由于答案可能很大，你只需输出这个数模P的值。

---

输入输出格式

 

第一行两个整数N和P(1≤P≤1000000000）。
第二行含有N个非负整数,从左到右依次为a1,a2,…,aN, (0≤ai≤1000000000,1≤i≤N)。
第三行有一个整数M，表示操作总数。
从第四行开始每行描述一个操作，输入的操作有以下三种形式：
操作1：“1 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai×c(1≤t≤g≤N,0≤c≤1000000000)。
操作2：“2 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai+c (1≤t≤g≤N,0≤c≤1000000000)。
操作3：“3 t g”(不含双引号)。询问所有满足t≤i≤g的ai的和模P的值 (1≤t≤g≤N)。
同一行相邻两数之间用一个空格隔开，每行开头和末尾没有多余空格。

---

输出格式：

 

对每个操作3，按照它在输入中出现的顺序，依次输出一行一个整数表示询问结果。

---

输入输出样例

输入样例#1：

7 43
1 2 3 4 5 6 7
5
1 2 5 5
3 2 4
2 3 7 9
3 1 3
3 4 7

输出样例#1：

2
35
8

说明

初始时数列为(1,2,3,4,5,6,7)。

经过第1次操作后，数列为(1,10,15,20,25,6,7)。

对第2次操作，和为10+15+20=45，模43的结果是2。

经过第3次操作后，数列为(1,10,24,29,34,15,16}

对第4次操作，和为1+10+24=35，模43的结果是35。

对第5次操作，和为29+34+15+16=94,模43的结果是8。

---

 

【题解】：

这个题目就是一个模板题。但是万事开头难，自己也错了不少次才AC的。

这个题目每一个节点不是以前的单纯记录Add[ ]即可。

还多了一个乘法的操作：

因为多了一个乘法的操作，所以必须增加一个mul[ ] 与之对应才行。

后来遇到问题了，又有乘法又有加法怎么弄呢？？？？

这个问题真的弄得头都大了。

后来瞄了一眼题解才知道，原来只要一直满足先乘法后加法才可以弄。

如果当前位置有乘法，想转移到子节点，

那么pushdown操作必须先乘后加。

首先当前节点 mul = k,add = c

k(ax+b)=ak x + bk

然后 ak *x + b*k+c

如果到了叶子节点那么就是直接把对应的位置算出即可。

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 100;
int n, m, L, R, Ins;
ll tree[N << 2], mul[N << 2], add[N << 2], a[N];
ll mod, c, tmp;
void Build(int No, int L, int R) {
    mul[No] = 1;
    int Mid = (L + R) >> 1;
    if (L == R) {
        tree[No] = a[L];
        return;
    }
    Build(No << 1, L, Mid);
    Build(No << 1 | 1, Mid + 1, R);
    tree[No] = (tree[No << 1] + tree[No << 1 | 1]) % mod;
}
void Mul(int No, int L, int R, ll k) {
    mul[No] = mul[No] * k % mod;
    add[No] = add[No] * k % mod;
    tree[No] = (tree[No] * k) % mod;
}
void Add(int No, int L, int R, ll k) {
    add[No] = (add[No] + k) % mod;
    tree[No] = (tree[No] + (R - L + 1) * k % mod) % mod;
}
void Push_down(int No, int L, int R) {
    int Mid = (L + R) >> 1;
    if (mul[No] == 1 && add[No] == 0)
        return;

    Mul(No << 1, L, Mid, mul[No]);
    Mul(No << 1 | 1, Mid + 1, R, mul[No]);

    Add(No << 1, L, Mid, add[No]);
    Add(No << 1 | 1, Mid + 1, R, add[No]);

    mul[No] = 1;
    add[No] = 0;
}
void modify(int No, int L, int R, int x, int y, ll k, int f) {
    if (x <= L && R <= y && f == 1) {
        tree[No] = tree[No] * k % mod;
        mul[No] = mul[No] * k % mod;
        add[No] = add[No] * k % mod;
        return;
    }
    if (x <= L && R <= y && f == 2) {
        tree[No] = (tree[No] + (R - L + 1) * k) % mod;
        add[No] = (add[No] + k) % mod;
        return;
    }
    int Mid = (L + R) >> 1;
    Push_down(No, L, R);
    if (x <= Mid)
        modify(No << 1, L, Mid, x, y, k, f);
    if (Mid < y)
        modify(No << 1 | 1, Mid + 1, R, x, y, k, f);
    tree[No] = (tree[No << 1] + tree[No << 1 | 1]) % mod;
}
ll query(int No, int L, int R, int x, int y) {
    if (x <= L && R <= y)
        return tree[No];
    int Mid = (L + R) >> 1;
    Push_down(No, L, R);
    ll res = 0;
    if (x <= Mid)
        res = (res + query(No << 1, L, Mid, x, y)) % mod;
    if (Mid < y)
        res = (res + query(No << 1 | 1, Mid + 1, R, x, y)) % mod;
    return res % mod;
}
void check(int No, int L, int R) {
    int Mid = (L + R) >> 1;
    if (L == R) {
        printf("(%d,%lld)\n", L, tree[No]);
        return;
    }
    check(No << 1, L, Mid);
    check(No << 1 | 1, Mid + 1, R);
}
int main() {
    scanf("%d%lld", &n, &mod);
    for (int i = 1; i <= n; i++) {
        scanf("%lld", &a[i]);
    }
    Build(1, 1, n);
    scanf("%d", &m);
    while (m--) {
        scanf("%d%d%d", &Ins, &L, &R);
        switch (Ins) {
            case 1:
                scanf("%lld", &c);
                modify(1, 1, n, L, R, c, 1);
                // check(1,1,n);
                break;
            case 2:
                scanf("%lld", &c);
                modify(1, 1, n, L, R, c, 2);
                // check(1,1,n);
                break;
            case 3:
                tmp = query(1, 1, n, L, R);
                printf("%lld\n", tmp);
                break;
        }
    }
    return 0;
}