洛谷线段树2

第一行包含三个整数 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 取模所得的结果

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

const int N = 2e5 + 10;
int tot = 0;
int n, m;
ll p;
struct node
{
    int l, r;
    ll sum, add, mul;
    node() {l = r = sum = add = 0; mul = 1;};
}tree[N * 4];

inline int ls(int cur) 
{
    return cur << 1;
}

inline int rs (int cur)
{
    return (cur << 1) | 1;
}

inline void push_up(int cur)
{
    tree[cur].sum = (tree[ls(cur)].sum + tree[rs(cur)].sum) % p;
}

inline void push_down(int cur)
{
    if (tree[cur].mul != 1 or tree[cur].add != 0) {
        tree[ls(cur)].sum = (tree[ls(cur)].sum * tree[cur].mul + (tree[ls(cur)].r - tree[ls(cur)].l + 1) * tree[cur].add) % p;
        tree[rs(cur)].sum = (tree[rs(cur)].sum * tree[cur].mul + (tree[rs(cur)].r - tree[rs(cur)].l + 1) * tree[cur].add) % p;
        tree[ls(cur)].mul = tree[ls(cur)].mul * tree[cur].mul % p;
        tree[rs(cur)].mul = tree[rs(cur)].mul * tree[cur].mul % p;
        tree[ls(cur)].add = (tree[ls(cur)].add * tree[cur].mul + tree[cur].add) % p;
        tree[rs(cur)].add = (tree[rs(cur)].add * tree[cur].mul + tree[cur].add) % p;
        tree[cur].add = 0;
        tree[cur].mul = 1;
    }
}


void build(int cur, int l, int r) 
{
    if (l == r) {
        ll a;
        scanf("%lld", &a);
        tree[cur].sum = a;
        tree[cur].l = l;
        tree[cur].r = r;
        return;
    }
    int mid = (l + r) >> 1;
    tree[cur].l = l; tree[cur].r = r;
    build(ls(cur), l, mid);
    build(rs(cur), mid + 1, r);
    push_up(cur);
}

void add(int cur, int l, int r, int v)
{
    if (tree[cur].l == l and tree[cur].r == r){
        tree[cur].sum = (tree[cur].sum + (r - l + 1) * v) %p;
        tree[cur].add = (tree[cur].add + v) % p;
        return;
    }
    push_down(cur);
    //继续往下
    //更新接下来的区间,保证他们是最新的
    int mid = (tree[cur].l + tree[cur].r) >> 1;
    if (r <= mid) {
        add(ls(cur), l, r, v);
    }
    else if (l > mid) {
        add(rs(cur), l, r, v);
    }
    else {
        add(ls(cur), l, mid, v);
        add(rs(cur), mid + 1, r, v);
    }
    //所有区间都从自己的子区间获取值的更新
    push_up(cur);
}

void mul(int cur, int l, int r, int v)
{
    if (tree[cur].l == l and tree[cur].r == r){
        tree[cur].sum = (tree[cur].sum * v) % p;
        tree[cur].mul = (tree[cur].mul * v) % p;
        tree[cur].add = (tree[cur].add * v) % p;
        return;
    }
    push_down(cur);
    int mid = (tree[cur].l + tree[cur].r) >> 1;
    if (r <= mid) {
        mul(ls(cur), l, r, v);
    }
    else if (l > mid) {
        mul(rs(cur), l, r, v);
    }
    else {
        mul(ls(cur), l, mid, v);
        mul(rs(cur), mid + 1, r, v);
    }
    //所有区间都从自己的子区间获取值的更新
    push_up(cur);
}

ll query(int cur, int l, int r)
{
    if (tree[cur].l == l and tree[cur].r == r){
        return tree[cur].sum;
    }
    push_down(cur);
    int mid = (tree[cur].l + tree[cur].r) >> 1;
    if (r <= mid) {
        return query(ls(cur), l, r);
    }
    else if (l > mid) {
        return query(rs(cur), l, r);
    }
    else {
        return (query(ls(cur), l, mid) + query(rs(cur), mid + 1, r)) % p;
    }
    //没有更新操作,所以不需要push_up
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("D:/VS CODE/C++/in.txt", "r", stdin);
    freopen("D:/VS CODE/C++/out.txt", "w", stdout);
#endif
    cin >> n >> m >> p;
    build(1, 1, n);
    while (m--) {
        int o, x, y, k;
        scanf("%d", &o);
        if (o == 1) {
            cin >> x >> y >> k;
            mul(1, x, y, k);
        }
        else if (o == 2) {
            cin >> x >> y >> k;
            add(1, x, y, k);
        }
        else {
            cin >> x >> y;
            cout << query(1, x, y) << endl;
        }
    }

    fclose(stdin);
    fclose(stdout);
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
洛谷P1168题目是关于中位数线段树解法的问题。中位数线段树解法可以通过维护两个堆来实现。一个是大根堆,一个是小根堆。每次插入元素时,根据一定的规则来维护这两个堆,使得大根堆的个数在一定情况下比小根堆多1或者相等。大根堆的最后一个元素即为中位数。具体的规则如下: 1. 如果大根堆和小根堆的个数相等,下一次插入的元素一定插入到大根堆。此时判断小根堆的堆顶是否大于当前元素x,如果是,则将小根堆的堆顶元素插入到大根堆,然后将x压入小根堆;否则直接将x压入大根堆。 2. 如果大根堆和小根堆的个数不相等,按照类似的规则进行操作。 通过以上规则,可以实现在每次插入元素时,维护两个堆的平衡,并且保证大根堆的最后一个元素即为中位数。 这种解法的时间复杂度为O(logN),其中N为序列的长度。 #### 引用[.reference_title] - *1* *2* [中位数(洛谷p1168)(堆/树状数组+二分/线段树+二分)](https://blog.csdn.net/qq_45604735/article/details/114382762)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [洛谷 P1168 中位数(权值线段树,离散化)](https://blog.csdn.net/qq_38232157/article/details/127594230)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值