洛谷P3373 线段树2(乘法加法lazytag)

线段树模板题,含lazytag的线段树码量本身就比较大,再加入乘法标记,还要考虑先乘后加的问题,本蒟蒻一调就是几个小时。

P3373 【模板】线段树 2icon-default.png?t=M4ADhttps://www.luogu.com.cn/problem/P3373

#include <bits/stdc++.h>
#define sc(n) scanf("%d", &n)
#define int long long
#define endl '\n'
#define FAST ios::sync_with_stdio(false);
typedef long long ll;
using namespace std;
int n, m, mod;
const int N = 1e5 + 10;
struct node
{
    int l, r, v, mul, add;
} t[5*N];
int a[N];
void build(int p, int l, int r)
{
    t[p].mul = 1;t[p].add = 0;t[p].l = l, t[p].r = r;
    if (l == r)
    {
        t[p].v = a[l]%mod;
        return;
    }
    int mid = l + r >> 1;
    build(p << 1, l, mid);
    build(p << 1 | 1, mid + 1, r);
    t[p].v = (t[p << 1].v + t[p << 1 | 1].v)%mod;
}
void push_down(int p)
{
    t[p << 1].v = (ll)(t[p].mul * t[p << 1].v + ((t[p << 1].r - t[p << 1].l + 1) * t[p].add)%mod)%mod;
    t[p << 1 | 1].v = (ll)(t[p].mul * t[p << 1 | 1].v%mod + ((t[p << 1 | 1].r - t[p << 1 | 1].l+1) * t[p].add)%mod)%mod;
    t[p << 1].mul = (ll)(t[p << 1].mul * t[p].mul) % mod;
    t[p << 1 | 1].mul = (ll)(t[p << 1 | 1].mul * t[p].mul)% mod;
    t[p << 1].add = (ll)((t[p << 1].add * t[p].mul)%mod+t[p].add) % mod;
    t[p << 1 | 1].add =(ll)((t[p << 1 | 1].add * t[p].mul)%mod+t[p].add) % mod;
    t[p].mul = 1, t[p].add = 0;
    return;
}

int query(int p, int l, int r)
{
    if (t[p].l >= l && t[p].r <= r)
    {
        return t[p].v;
    }
    push_down(p);
    int res = 0;
    if (t[p<<1].r >= l)
        res = (res+query(p << 1, l, r))%mod;
    if (t[p<<1|1].l <= r)
        res = (res+query(p << 1 | 1, l, r))%mod;
    return res;
}
void add(int p, int l, int r, int val)
{
    if (t[p].l >= l && t[p].r <= r)
    {
        t[p].v = (t[p].v+((t[p].r - t[p].l + 1) * val)%mod)% mod;
        t[p].add =(val+t[p].add)% mod;
        return;
    }
    push_down(p);
    int mid = l + r >> 1;
    if (t[p<<1].r >= l)
        add(p << 1, l, r, val);
    if (t[p<<1|1].l <= r)
        add(p << 1 | 1, l, r, val);
    t[p].v = (t[p << 1].v + t[p << 1 | 1].v)%mod;
}
void mul(int p, int l, int r, int val)
{
    if (t[p].l >= l && t[p].r <= r)
    {
        t[p].v =(t[p].v*val) % mod;
        t[p].mul = (t[p].mul*val) % mod;
        t[p].add = (t[p].add*val) % mod;
        return;
    }
    push_down(p);
    int mid = l + r >> 1;
    if (t[p<<1].r >= l)
        mul(p << 1, l, r, val);
    if (t[p<<1|1].l <= r)
        mul(p << 1 | 1, l, r, val);
    t[p].v = (t[p << 1].v + t[p << 1 | 1].v)%mod;
}
signed main()
{
    FAST;
    cin >> n >> m >> mod;
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    build(1, 1, n);
    int op, x, y;
    while (m--)
    {
        cin >> op;
        if (op == 1)
        {
            int k;
            cin >> x >> y >> k;
            mul(1, x, y, k);
        }
        else if (op == 2)
        {
            int k;
            cin >> x >> y >> k;
            add(1, x, y, k);
        }
        else if (op == 3)
        {
            cin >> x >> y;
            cout << query(1, x, y) << endl;
        }
    }
    return 0;
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述 有 $n$ 个点在数轴上,每个点有一个权值 $a_i$,你需要支持以下操作: - 修改一个点的权值。 - 给出 $l,r,k$,询问在区间 $[l,r]$ 中,权值严格大于 $k$ 的点的个数。 输入格式 第一行一个正整数 $n(1\leq n\leq 5\times10^5)$。 第二行 $n$ 个整数 $a_i(|a_i|\leq 10^9)$,表示每个点的权值。 第三行一个正整数 $m(1\leq m\leq 5\times10^5)$。 接下来 $m$ 行,每行一个操作,格式如下: - “Q l r k” 表示询问区间 $[l,r]$ 中,权值严格大于 $k$ 的点的个数。 - “C x y” 表示将第 $x$ 个点的权值修改为 $y$。 输出格式 对于每个询问操作,输出其结果。 输入样例 5 0 1 2 3 4 4 Q 2 5 3 C 4 6 Q 1 5 2 Q 3 4 4 输出样例 1 2 0 算法1 线段(动态开点) 线段的思想是把区间分成若干个小区间,每个小区间对应一段线段。对于每个线段,维护一些信息,例如区间和、区间最大值等等。 对于这道题目,我们可以按照值域线段的思想,将区间对应到值域上。即将整个区间 $[0,n-1]$ 对应到值域上,建立一棵值域线段。对于线段上的每个节点,维护该节点对应的区间内权值大于某个值 $k$ 的点的个数。当然,对于叶子节点,该值就是 $0$ 或 $1$。 对于一个查询操作 $Q(l,r,k)$,需要在值域线段上找到 $[l,r]$ 对应的区间,然后查询该区间内权值大于 $k$ 的点的个数。这个可以通过线段的区间查询操作实现。 对于一个修改操作 $C(x,y)$,需要在值域线段上找到 $x$ 对应的叶子节点,然后修改该叶子节点的值为 $y$,然后向上更新整个线段,直到根节点。 时间复杂度 对于每次修改和查询操作,都需要在值域线段上查询或修改,时间复杂度是 $O(\log n)$。总时间复杂度是 $O(m\log n)$。 C++ 代码

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值