题目链接: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的数的迭代器
代码部分:
#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;
}