标签：return MOBAN rs int 线段 mid 分裂 ls sm

线段树分裂线段树合并板子LUOGU5494

#include <bits/stdc++.h>

using namespace std;
const int maxn = 2e5 + 5;
typedef long long ll;
int ls[maxn * 60], rs[maxn * 60], tot;
int pool[maxn * 60], lj;
int newnode() { return lj ? pool[lj--] : ++tot; }

ll sm[maxn * 60];
int n, m;
int rt[maxn], nowrt = 1;
void del(int p) {
    pool[++lj] = p;
    ls[p] = rs[p] = sm[p] = 0;
}
void maketree(int &p, int l, int r) {
    p = ++tot;
    if (l == r) {
        scanf("%lld", &sm[p]);
        return;
    }
    int mid = (l + r) >> 1;
    maketree(ls[p], l, mid);
    maketree(rs[p], mid + 1, r);
    sm[p] = sm[ls[p]] + sm[rs[p]];
}
void ins(int &p, int l, int r, int num, int x) {
    if (!p)
        p = newnode();
    sm[p] += num;
    if (l == r) {
        return;
    }

    int mid = (l + r) >> 1;
    if (x <= mid)
        ins(ls[p], l, mid, num, x);
    else
        ins(rs[p], mid + 1, r, num, x);
}
int merge(int x, int y) {
    if ((!x) || (!y))
        return x + y;
    ls[x] = merge(ls[x], ls[y]);
    rs[x] = merge(rs[x], rs[y]);
    sm[x] += sm[y];
    del(y);
    return x;
}
void split(int x, int &y, ll k) {
    if (x == 0)
        return;
    y = newnode();
    ll lsz = sm[ls[x]];
    if (k > lsz) {
        split(rs[x], rs[y], k - lsz);
    } else {
        rs[y] = rs[x];
        rs[x] = 0;
    }
    if (k < lsz) {
        split(ls[x], ls[y], k);
    }
    sm[y] = sm[x] - k;
    sm[x] = k;
    return;
}
ll query(int &p, int l, int r, int x, int y) {
    if (!p)
        return 0;
    if (x <= l && r <= y) {
        return sm[p];
    }
    int mid = (l + r) >> 1;
    if (y <= mid)
        return query(ls[p], l, mid, x, y);
    else if (x > mid)
        return query(rs[p], mid + 1, r, x, y);
    else
        return query(ls[p], l, mid, x, y) + query(rs[p], mid + 1, r, x, y);
}
int getk(int p, int l, int r, int k) {
    if (l == r)
        return l;
    int mid = (l + r) >> 1;
    if (k > sm[ls[p]])
        return getk(rs[p], mid + 1, r, k - sm[ls[p]]);
    else
        return getk(ls[p], l, mid, k);
}
int main() {
    scanf("%d%d", &n, &m);
    maketree(rt[1], 1, n);
    int opt, p, x, y, t, q, k;
    for (int i = 1; i <= m; i++) {
        scanf("%d", &opt);
        if (opt == 0) {
            scanf("%d%d%d", &p, &x, &y);
            ll sz1 = query(rt[p], 1, n, 1, y);
            ll sz2 = query(rt[p], 1, n, x, y);
            ++nowrt;
            split(rt[p], rt[nowrt], sz1 - sz2);
            int tmprt = 0;
            split(rt[nowrt], tmprt, sz2);
            rt[p] = merge(rt[p], tmprt);
        } else if (opt == 1) {
            scanf("%d%d", &p, &t);
            rt[p] = merge(rt[p], rt[t]);
        } else if (opt == 2) {
            scanf("%d%d%d", &p, &x, &q);
            ins(rt[p], 1, n, x, q);
        } else if (opt == 3) {
            scanf("%d%d%d", &p, &x, &y);
            printf("%lld\n", query(rt[p], 1, n, x, y));
        } else {
            scanf("%d%d", &p, &k);
            if (sm[rt[p]] < k)
                puts("-1");
            else {
                printf("%d\n", getk(rt[p], 1, n, k));
            }
        }
    }
    return 0;
}

标签：return,MOBAN,rs,int,线段,mid,分裂,ls,sm
来源： https://www.cnblogs.com/newuser233/p/segmentmergesplit.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。