线段树区间维护（加乘题解）洛谷3373

题目描述

如题，已知一个数列，你需要进行下面三种操作：

• 将某区间每一个数乘上 x；
• 将某区间每一个数加上 x；
• 求出某区间每一个数的和。

输入格式

第一行包含三个整数 n,q,m，分别表示该数列数字的个数、操作的总个数和模数。

第二行包含 n 个用空格分隔的整数，其中第 i 个数字表示数列第 i 项的初始值。

接下来 q 行每行包含若干个整数，表示一个操作，具体如下：

操作 11： 格式：1 x y k 含义：将区间 [x,y] 内每个数乘上 k

操作 22： 格式：2 x y k 含义：将区间 [x,y] 内每个数加上 k

操作 33： 格式：3 x y 含义：输出区间 [x,y] 内每个数的和对m 取模所得的结果

输出格式

输出包含若干行整数，即为所有操作 33 的结果。

输入输出样例

输入 #1复制

5 5 38
1 5 4 2 3
2 1 4 1
3 2 5
1 2 4 2
2 3 5 5
3 1 4

输出 #1复制

17
2

思路：

弄两个lazy标记，tag1(记录加法),tag2（记录乘法）。

举个例子：a,c1,c2,c3,s1,s2,s3.

a是原数，s1,s2,s3是要乘的，c1,c2,c3是要加的。

对于((a+c1)*s1+c2)*s2*s3+c3

结果为a*s1*s2*s3+c1*s1*s2*s3+c2*s2*s3+c3.

可以看看得出规律：对于每次运算，如果是加法运算则让tag1加一下，对于乘法运算，则tag1,tag2都乘一下，结果为，a*tag2+tag1。

代码

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
using namespace std;
const int N = 1e5 +100;
#define ls (o<<1)
#define rs (o<<1|1)
long long  sum[N << 2];
long long  tag[N << 2], tag2[N << 2];
long long  aa[N];
int  k, a, b, m, c, n, x;
long long p;
void build(int o, int l, int r)
{
    if (l == r)
    {
        sum[o] = aa[l];
        return;
    }
    int mid = (l + r) >> 1;
    build(ls, l, mid);
    build(rs, mid + 1, r);
    sum[o] = sum[ls] + sum[rs];
}
void spread(int o, int l, int r)
{
    if (tag2[o] != 1)
    {
        sum[ls] = (sum[ls] * tag2[o]) % p;
        sum[rs] = (sum[rs] * tag2[o]) % p;
        tag2[ls] = (tag2[ls] * tag2[o]) % p;
        tag2[rs] = (tag2[rs] * tag2[o]) % p;
        tag[ls] = (tag[ls] * tag2[o]) % p;
        tag[rs] = (tag[rs] * tag2[o]) % p;
        tag2[o] = 1;
    }
    if (tag[o])
    {
        int mid = (l + r) >> 1;
        sum[ls] = (sum[ls] + tag[o] * (mid - l + 1)) % p;
        sum[rs] = (sum[rs] + tag[o] * (r - mid)) % p;
        tag[ls] = (tag[ls] + tag[o]) % p;
        tag[rs] = (tag[rs] + tag[o]) % p;
        tag[o] = 0;
    }
}
void change(int o, int l, int r, int d, int ql, int qr, int y)
{
    if (l > qr || r < ql) return;
    if (ql <= l && r <= qr)
    {
        if (y == 2)
        {
            sum[o] = (sum[o] + d * (r - l + 1)) % p;
            tag[o] = (tag[o] + d) % p;
        }
        else
        {
            sum[o] = (d * sum[o]) % p;
            tag[o] = (tag[o] * d) % p;
            tag2[o] = (tag2[o] * d) % p;
        }
        return;
    }
    spread(o, l, r);
    int mid = (l + r) >> 1;
    if (ql <= mid)
        change(ls, l, mid, d, ql, qr, y);
    if (mid + 1 <= qr)
        change(rs, mid + 1, r, d, ql, qr, y);
    sum[o] = (sum[ls] + sum[rs]) % p;
}
long long  ask(int o, int l, int r, int ql, int qr)
{
    if (ql <= l && r <= qr)
        return sum[o] % p;
    spread(o, l, r);
    int mid = (l + r) >> 1;
    long long ans = 0;
    if (ql <= mid)
        ans = (ans + ask(ls, l, mid, ql, qr)) % p;
    if (mid + 1 <= qr)
        ans = (ans + ask(rs, mid + 1, r, ql, qr)) % p;
    return ans;
}
int main()
{
    cin >> n >>m>>p;
    for (int i = 1; i <= n; i++)
        scanf("%lld", &aa[i]);
    build(1, 1, n);
    for (int i = 1; i <= 4 * n + 100; i++)
        tag2[i] = 1;
    for (int i = 1; i <= m; i++)
    {
        int y;
        scanf("%d", &y);
        if (y == 3)
        {
            scanf("%d%d", &a, &b);
            printf("%lld\n", (ask(1, 1, n, a, b) % p + p) % p);
        }
        else
        {
            scanf("%d%d%d", &a, &b, &c);
            change(1, 1, n, c, a, b, y);
        }
    }
    return 0;
}