题目链接 :P5076 【深基16.例7】普通二叉树(简化版) - 洛谷 | 计算机科学教育新生态
题目难度 :普及+/提高
解题思路 :本题运用了STL 中的multiset ,它可以看成一个序列,插入一个数,删除一个数都能够在O(logn)的时间内完成,而且他能时刻保证序列中的数是有序的,而且序列中可以存在重复的数(区别于set是不能有重复元素), 首先解决这道题用到了lower_bound(返回元素值为X的第一个可安插位置,也就是元素值 >=X的第一个元素位置)和upper_down(返回元素值为X 的最后一个可安插位置,也就是元素值 > X 的第一个元素位置)函数,还有multiset的迭代器(multiset<int>::iterator)
常用函数:
multiset<int>q;
//定义一个multiset,尖括号里写类型
//如果是自定义类型,需要重载小于号
q.insert(x);
//插入一个数 x
q.clear();
//清空
q.erase(x);
//删除容器中的所有值为 x 的数
q.erase(it);
//删除容器中迭代器it指向的元素
q.empty();
//返回bool值,如果容器为空返回true,否则返回false
q.size()
//返回元素个数
q.begin();
//返回首个元素的迭代器
q.end();
//返回最后一个元素的下一个位置的迭代器
q.count(x);
//返回容器中 x 的个数
q.find(x);
//返回容器中第一个x的位置(迭代器),如果没有就返回q.end()
q.lower_bound(x);
//返回容器中第一个大于等于x的数的迭代器
q.upper_bound(x);
//返回容器中第一个大于x的数的迭代器
代码部分:
cpp
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 110;
multiset<int>m;
int q,op,x;
int rank1;//记录该数的排位
int main() {
ios::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> q;
m.insert(-0x7fffffff);
m.insert(0x7fffffff);//提前放入这两个数
while(q--)
{
cin >> op >> x;
if(op == 1)
{
auto now = m.lower_bound(x);//返回迭代器 now取得是x的位置
//可以写作multiset<int>::iterator,因为lower_bound方法返回的是迭代器
rank1 = 0;
for(auto i = m.begin(); i != now; i++,rank1++);//切记要加分号
cout<<rank1<<'\n';//输出排名
}
else if(op == 2)
{
rank1 = -1;//前面还有个0x7f7f7f7f7
for(int i:m)
{
if(++rank1 == x) cout<<i<<'\n';
}
/*也可以这样遍历
for(multiset<int>::iterator i=m.begin();i!=m.end();i++)
{
rank1++;
if(rank1==x)
cout<<i<<'\n';
}
*/
}
else if(op == 3)
{
auto now = m.lower_bound(x);
cout << *--now <<'\n'; //因为是迭代器(指针),所以输出前面加 *
//由于我们要取得前驱,所以now要自减一
}
else if(op == 4)
{
cout<<*m.upper_bound(x)<<'\n'; //因为是迭代器(指针),所以输出前面加 *
}
else
{
m.insert(x);
}
}
return 0;
}