zoukankan      html  css  js  c++  java
  • P3373 【模板】线段树 2(板子好题)

    题目描述

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

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

    输入格式

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

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

    接下来 mmm 行每行包含若干个整数,表示一个操作,具体如下:

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

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

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

    输出格式

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

    输入输出样例

    输入 #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 tag,而关键就是确定加法和乘法的优先级。这里举一个例子:

    一、如果是先加后乘的话,对于某一个节点a[i]:

    1)+2:  sum = a[i] + 2

    2) *3:  sum = (a[i] + 2) * 3

    这时一切都很正常,但如果接下来再执行+2操作,期望的结果是sum = (a[i] + 2) * 3 + 2,然而得到的却是sum = (a[i] + 2 + 2) * 3, 因为lazy tag的效果是累加的。因此需要使+ 2变成+ 2 / 3, 扩展到了实数域会带来精度损失,这就是先加后乘的弊端。

    一、如果是先乘后加的话,对于某一个节点a[i]:

    1)+2:  sum = a[i] + 2

    2) *3:  sum = a[i] * 3 + 2 * 3

    3)+2:  sum = a[i] * 3 + 2 * 3 + 2

    4) *4:  sum = a[i] * 3 * 4 + 2 * 3 * 4 + 2 * 4

    看起来很完美,因此选择先乘后加。这样需要注意一下change函数和spread函数。change的时候,如果是进行区间加法则正常进行,如果是进行区间乘法,则记得给加法标记也乘一下k。对于spread函数,要按照:更新sum值-更新乘法标记-更新加法标记-父节点清除加法标记乘法标记的顺序写。

    需要注意的地方:建树时初始化乘法标记为1,清除标记时乘法标记清除为1.

    不要多加冗余的取模运算,会T

    #include <bits/stdc++.h>
    #define SIZE 100005
    #define int long long
    using namespace std;
    int a[SIZE], n, q, mod;
    inline int read(){
        int a = 0; int f = 0;char p = getchar();
        while(!isdigit(p)){f |= p == '-'; p = getchar();}
        while(isdigit(p)){a = (a << 3) + (a << 1) + (p ^ 48); p = getchar();}
        return f ? -a : a;
    }
    struct SegmentTree
    {
        int p;
        int l;
        int r;
        long long sum;
        long long add;
        long long mult;
    } t[4*SIZE];
    void build(int p, int l, int r)
    {
        t[p].l = l, t[p].r = r, t[p].mult = 1;//注意初始化mult 
        if(l == r)    {    t[p].sum = 1ll * a[l] % mod; return;    }//注意build函数 
        int mid = (l + r) >> 1;
        build(2 * p, l, mid);
        build(2 * p + 1, mid + 1, r);
        t[p].sum = 1ll * (t[2 * p].sum + t[2 * p + 1].sum) % mod;
    }
    void spread(int p)
    {
        t[2 * p].sum = 1ll * (t[p].mult * t[2 * p].sum + t[p].add * (t[2 * p].r - t[2 * p].l + 1) % mod) % mod;//此处父节点的加法标记已经乘过了,因此也满足先乘后加 
        t[2 * p + 1].sum = 1ll * (t[p].mult * t[2 * p + 1].sum % mod + t[p].add * (t[2 * p + 1].r - t[2 * p + 1].l + 1) % mod) % mod;
        t[2 * p].mult = 1ll * t[2 * p].mult * t[p].mult % mod;
        t[2 * p + 1].mult = 1ll * t[2 * p + 1].mult * t[p].mult % mod;
        t[2 * p].add = 1ll * (t[p].add + t[2 * p].add * t[p].mult) % mod;
        t[2 * p + 1].add = 1ll * (t[p].add + t[2 * p + 1].add * t[p].mult % mod) % mod;
        t[p].add = 0;
        t[p].mult = 1;
    }
    void change(int op, int p, int l, int r, int k)
    {
        if(t[p].l >= l && t[p].r <= r)//完全覆盖 
        {
            if(op == 1)
            {
                t[p].add = 1ll * t[p].add * k % mod;
                t[p].sum = 1ll * (t[p].sum * k) % mod;
                t[p].mult = 1ll * t[p].mult * k % mod;
                return;
            }
            else
            {
                t[p].sum = 1ll * (t[p].sum + (long long)k * (t[p].r - t[p].l + 1) ) % mod;
                t[p].add = 1ll * (t[p].add + k) % mod;
                return;
            }
        }
        spread(p);
        t[p].sum = 1ll * (t[2 * p].sum  + t[2 * p + 1].sum ) % mod;
        int mid = (t[p].l + t[p].r) >> 1;
        if(l <= mid) change(op, 2 * p, l, r, k);
        if(r > mid) change(op, 2 * p + 1, l, r, k);
        t[p].sum = 1ll * (t[2 * p].sum + t[2 * p + 1].sum) % mod;
    }
    long long ask(int p, int l, int r)
    {
        if(t[p].l >= l && t[p].r <= r) return t[p].sum;
        spread(p);
        int mid = (t[p].l + t[p].r) >> 1;
        long long val = 0;
        if(l <= mid) val = 1ll * (val + ask(2 * p, l, r)) % mod;
        if(r > mid) val = 1ll * (val + ask(2 * p + 1, l, r)) % mod;
        return val;
    }
    signed main()
    {
        cin >> n >> q >> mod;
        int i, op, x, y, k;
        for(i = 1; i <= n; i++) a[i] = read();
        build(1, 1, n);
        for(i = 1; i <= q; i++)
        {
            op = read(), x = read(), y = read();
            if(op == 1)
            {
                k = read();
                change(1, 1, x, y, k);
            }
            else if(op == 2)
            {
                k = read();
                change(2, 1, x, y, k);
            }
            else cout << ask(1, x, y) << endl;
        }
        return 0;
    }
  • 相关阅读:
    oop klass

    广义表
    Huffman树
    二叉搜索树
    二叉树的前序、中序、后序、层序遍历
    循环链表解决约瑟夫环问题
    搭建局域网SVN代码服务器
    【CheckList】精简用例,提升执行效率,减少漏测(总结篇)
    测试资源不同时,如何有针对性的设计测试用例?
  • 原文地址:https://www.cnblogs.com/lipoicyclic/p/13265829.html
Copyright © 2011-2022 走看看