AcWing 253. 普通平衡树

您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:

  1. 插入数值x。
  2. 删除数值x(若有多个相同的数,应只删除一个)。
  3. 查询数值x的排名(若有多个相同的数,应输出最小的排名)。
  4. 查询排名为x的数值。
  5. 求数值x的前驱(前驱定义为小于x的最大的数)。
  6. 求数值x的后继(后继定义为大于x的最小的数)。
    注意: 数据保证查询的结果一定存在。
    本题为平衡树基本操作的代码实现
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>using namespace std;const int N = 100010, INF = 1e8;int n;
struct Node
{int l, r;int key, val;int cnt, size;
}tr[N];int root, idx;void pushup(int p)
{tr[p].size = tr[tr[p].l].size + tr[tr[p].r].size + tr[p].cnt;
}int get_node(int key)
{tr[ ++ idx].key = key;tr[idx].val = rand();tr[idx].cnt = tr[idx].size = 1;return idx;
}void zig(int &p)    // 右旋
{int q = tr[p].l;tr[p].l = tr[q].r, tr[q].r = p, p = q;pushup(tr[p].r), pushup(p);
}void zag(int &p)    // 左旋
{int q = tr[p].r;tr[p].r = tr[q].l, tr[q].l = p, p = q;pushup(tr[p].l), pushup(p);
}void build()
{get_node(-INF), get_node(INF);root = 1, tr[1].r = 2;pushup(root);if (tr[1].val < tr[2].val) zag(root);
}void insert(int &p, int key)
{if (!p) p = get_node(key);else if (tr[p].key == key) tr[p].cnt ++ ;else if (tr[p].key > key){insert(tr[p].l, key);if (tr[tr[p].l].val > tr[p].val) zig(p);}else{insert(tr[p].r, key);if (tr[tr[p].r].val > tr[p].val) zag(p);}pushup(p);
}void remove(int &p, int key)
{if (!p) return;if (tr[p].key == key){if (tr[p].cnt > 1) tr[p].cnt -- ;else if (tr[p].l || tr[p].r){if (!tr[p].r || tr[tr[p].l].val > tr[tr[p].r].val){zig(p);remove(tr[p].r, key);}else{zag(p);remove(tr[p].l, key);}}else p = 0;}else if (tr[p].key > key) remove(tr[p].l, key);else remove(tr[p].r, key);pushup(p);
}int get_rank_by_key(int p, int key)    // 通过数值找排名
{if (!p) return 0;   // 本题中不会发生此情况if (tr[p].key == key) return tr[tr[p].l].size + 1;if (tr[p].key > key) return get_rank_by_key(tr[p].l, key);return tr[tr[p].l].size + tr[p].cnt + get_rank_by_key(tr[p].r, key);
}int get_key_by_rank(int p, int rank)   // 通过排名找数值
{if (!p) return INF;     // 本题中不会发生此情况if (tr[tr[p].l].size >= rank) return get_key_by_rank(tr[p].l, rank);if (tr[tr[p].l].size + tr[p].cnt >= rank) return tr[p].key;return get_key_by_rank(tr[p].r, rank - tr[tr[p].l].size - tr[p].cnt);
}int get_prev(int p, int key)   // 找到严格小于key的最大数
{if (!p) return -INF;if (tr[p].key >= key) return get_prev(tr[p].l, key);return max(tr[p].key, get_prev(tr[p].r, key));
}int get_next(int p, int key)    // 找到严格大于key的最小数
{if (!p) return INF;if (tr[p].key <= key) return get_next(tr[p].r, key);return min(tr[p].key, get_next(tr[p].l, key));
}int main()
{build();scanf("%d", &n);while (n -- ){int opt, x;scanf("%d%d", &opt, &x);if (opt == 1) insert(root, x);else if (opt == 2) remove(root, x);else if (opt == 3) printf("%d\n", get_rank_by_key(root, x) - 1);else if (opt == 4) printf("%d\n", get_key_by_rank(root, x + 1));else if (opt == 5) printf("%d\n", get_prev(root, x));else printf("%d\n", get_next(root, x));}return 0;
}

Vector版偷懒平衡树

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
vector<int>v;
int n,opt,x;
int main()
{v.reserve(100001);scanf("%d",&n);while(n--){scanf("%d%d",&opt,&x);if(opt==1)    v.insert(lower_bound(v.begin(),v.end(),x),x);if(opt==2)    v.erase (lower_bound(v.begin(),v.end(),x));if(opt==3)    printf("%d\n",lower_bound(v.begin(),v.end(),x)-v.begin()+1);if(opt==4)    printf("%d\n",v[x-1]);if(opt==5)    printf("%d\n",v[lower_bound(v.begin(),v.end(),x)-v.begin()-1]);if(opt==6)    printf("%d\n",v[upper_bound(v.begin(),v.end(),x)-v.begin()]);}return 0;
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/318076.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

.NET Core实战项目之CMS 第十三章 开发篇-在MVC项目结构介绍及应用第三方UI

作为后端开发的我来说&#xff0c;前端表示真心玩不转&#xff0c;你如果让我微调一个位置的样式的话还行&#xff0c;但是让我写一个很漂亮的后台的话&#xff0c;真心做不到&#xff0c;所以我一般会选择套用一些开源UI模板来进行系统UI的设计。那如何套用呢&#xff1f;今天…

YBTOJ:伞兵空降(二分图匹配)

文章目录题目描述解析代码题目描述 有n个点和m条边的有向无环图&#xff0c;在这张图上的某些点上空投伞兵&#xff0c;使伞兵可以走到图上所有的点。 且每个点只能被一个伞兵走一次。问至少需要放多少伞兵。 解析 考虑一开始给每个点分配一个伞兵&#xff0c;最差就是这样n…

CF388C-Fox and Card Game【博弈论,结论】

正题 题目链接:https://www.luogu.com.cn/problem/CF388C 题目大意 有nnn堆卡片&#xff0c;第iii堆有sis_isi​张&#xff0c;给出每张卡的权值。现在先手选择一堆取走堆底的牌&#xff0c;然后后手选择一堆取走堆顶的牌&#xff0c;直到所有牌被取走。在双方都要求最大化取…

[FFT/IFFT]快速傅里叶(逆)变化 + 递归和递推模板

现在时间是2021-2-2&#xff0c;重新回来看2019学习的一知半解的FFTFFTFFT&#xff0c;又有了新的理解 所以修改了以往写过的文章&#xff0c;并增添些许内容 因为过去一年多&#xff0c;上了高中&#xff0c;学的知识多了些&#xff0c;以前不懂的有些东西现在看来挺简单的&am…

软件开发模式:瀑布与敏捷

瀑布和敏捷不是什么新概念&#xff0c;这里只是个人在团队合作中不得不去思考而做的归纳和总结&#xff0c;同时记录自己曾经踩过的坑&#xff0c;新瓶装旧酒&#xff0c;希望对你有所启发。瀑布模式瀑布模型是比较传统一种开发模式&#xff0c;特别是在2B的传统企业&#xff0…

257. 关押罪犯

题意&#xff1a; S城现有两座监狱&#xff0c;一共关押着N名罪犯&#xff0c;编号分别为1~N。他们之间的关系自然也极不和谐。很多罪犯之间甚至积怨已久&#xff0c;如果客观条件具备则随时可能爆发冲突。我们用“怨气值”&#xff08;一个正整数值&#xff09;来表示某两名罪…

AC 自动机

比我小一届却吊打我的大脚玩家(djwj233)的博客 什么是 AC 自动机 AC 自动机是一种多模匹配算法&#xff0c;就是解决 多个模式串 匹配 单个/多个 文本串用的。 AC 自动机的过程 P3808 【模板】AC 自动机(简单版) 总的来说&#xff0c;AC 自动机类似将所有串跑一个 KMP。 看到有…

YBTOJ洛谷P3231:消毒(二分图匹配)

文章目录题目描述解析代码题目描述 最近在生物实验室工作的小 T 遇到了大麻烦。 由于实验室最近升级的缘故&#xff0c;他的分格实验皿是一个长方体&#xff0c;其尺寸为 a∗b∗ca*b*ca∗b∗c。为了实验的方便&#xff0c;它被划分为 a∗b∗ca*b*ca∗b∗c 个单位立方体区域&am…

CF388D-Fox and Perfect Sets【dp,线性基】

正题 题目链接:https://www.luogu.com.cn/problem/CF388D 题目大意 给出kkk&#xff0c;求有多少个集合SSS满足S⊆[1,k]S\sube [1,k]S⊆[1,k]且 a∈S,b∈S⇒axorb∈Sa\in S,b\in S\Rightarrow a\ xor\ b\in Sa∈S,b∈S⇒a xor b∈S 1≤k≤1091\leq k\leq 10^91≤k≤109 解题思…

503. 借教室

503. 借教室 题意&#xff1a; 在大学期间&#xff0c;经常需要租借教室。 大到院系举办活动&#xff0c;小到学习小组自习讨论&#xff0c;都需要向学校申请借教室。 教室的大小功能不同&#xff0c;借教室人的身份不同&#xff0c;借教室的手续也不一样。 面对海量租借教…

.net core+Spring Cloud学习之路 一

文章开头唠叨两句。2019年了&#xff0c;而自己参加工作也两年有余了&#xff0c;用一个词来概括这两年多的生活&#xff0c;就是&#xff1a;“碌碌无为”。也不能说一点收获都没有&#xff0c;但是很少。2019来了&#xff0c;我立志要打破现状&#xff0c;改变自己&#xff0…

CF1153F-Serval and Bonus Problem【dp,数学期望】

正题 题目链接:https://www.luogu.com.cn/problem/CF1153F 题目大意 在有nnn个区间的左右端点在[0,l)[0,l)[0,l)范围内随机&#xff0c;求被至少kkk个区间覆盖的期望长度。 1≤n,k≤2000,1≤l≤1091\leq n,k\leq 2000,1\leq l\leq 10^91≤n,k≤2000,1≤l≤109 解题思路 长度…

网络流优化:-1优化与当前弧优化

所谓网络流优化&#xff0c;就是对网络流算法进行优化 &#xff08;逃&#xff09; -1优化 大概就是如果在一次bfs搜出的图中发现当前这个点啥都增广不出来&#xff0c;就暂时把这个点扣掉 当前弧优化 在一次bfs搜出的图中&#xff0c;如果某条边已经搜过&#xff0c;就不必…

P3128 [USACO15DEC]Max Flow P

P3128 [USACO15DEC]Max Flow P 树上差分之点差分模板题 题目描述&#xff1a; FJ给他的牛棚的N(2≤N≤50,000)个隔间之间安装了N-1根管道&#xff0c;隔间编号从1到N。所有隔间都被管道连通了。 FJ有K(1≤K≤100,000)条运输牛奶的路线&#xff0c;第i条路线从隔间si运输到隔…

插头DP/轮廓线DP

题解 P5056 【【模板】插头dp】- GNAQ (\(\uparrow\) 学习资料&#xff0c;大部分贺的&#xff0c;有一些些的改动与自己的补充) 什么是插头 DP 插头 DP 是一类用状压 DP 来处理连通性问题的 DP 方法。 常见的类型&#xff1a;棋盘插头 DP、连通性问题(回路问题&#xff0c;路径…

周末狂欢赛4(1-02E. JM的西伯利亚特快专递,寿司晚宴,荷马史诗)

文章目录T1&#xff1a;1-02E. JM的西伯利亚特快专递题目题解codeT2&#xff1a;寿司晚宴题目题解codeT3&#xff1a;荷马史诗题目题解codeT1&#xff1a;1-02E. JM的西伯利亚特快专递 题目 今天JM收到了一份来自西伯利亚的特快专递&#xff0c;里面装了一个字符串 s &#x…

.NET Core容器化开发系列(一)——Docker里面跑个.NET Core

前言博客园中已经有很多如何在Docker里面运行ASP.NET Core的介绍了。本篇主要介绍一些细节&#xff0c;帮助初学的朋友更加深入地理解如何在Docker中运行ASP.NET Core。安装DockerDocker现支持在主流Linux、Windows和macOS上安装&#xff0c;官方的安装文档请参考docker docs。…

YBTOJ:卖猪问题(网络流)

文章目录题目描述数据范围解析代码题目描述 尼克在一家养猪场工作&#xff0c;这家养猪场共有MMM间锁起来的猪舍&#xff0c;由于猪舍的钥匙都给了客户&#xff0c;所以尼克没有办法打开这些猪舍。有NNN个客户从早上开始一个接一个来购买生猪&#xff0c;他们到达后首先用手中…

P4370-[Code+#4]组合数问题2【数学,堆】

正题 题目链接:https://www.luogu.com.cn/problem/P4370 题目大意 求满足m≤n≤am\leq n\leq am≤n≤a的情况下&#xff0c;前kkk大的(nm)\binom{n}{m}(mn​)的和。 1≤n≤106,1≤k≤1051\leq n\leq 10^6,1\leq k\leq 10^51≤n≤106,1≤k≤105 解题思路 首先想到的是(nm)>…

【做题记录】[SCOI2009]围豆豆

[SCOI2009]围豆豆 \(n\times m(n,m\le 10)\) 的网格中有 \(d\) 个球 \((d\le 9)\)&#xff0c;要求在网格中选定一个起点开始做一个欧拉回路&#xff0c;路径的价值为路径完全包住的球的价值之和减去路径长度&#xff0c;求所有路径中的价值最大值。 有价值与步数的两个限制&am…