【深基16.例7】普通二叉樹(簡化版)
題目:【深基16.例7】普通二叉樹(簡化版)
題目連結:https://www.luogu.com.cn/problem/T237469?contestId=68358
題目描述
您需要寫一種資料結構,來維護一些數( 都是 10^9109 以內的數字)的集合,最開始時集合是空的。其中需要提供以下操作,操作次數 qq 不超過 10^4104:
- 查詢 xx 數的排名(排名定義為比當前數小的數的個數 +1+1。若有多個相同的數,應輸出最小的排名)。
- 查詢排名為 xx 的數。
- 求 xx 的前驅(前驅定義為小於 xx,且最大的數)。若未找到則輸出 -2147483647−2147483647。
- 求 xx 的後繼(後繼定義為大於 xx,且最小的數)。若未找到則輸出 21474836472147483647。
- 插入一個數 xx。
輸入格式
無
輸出格式
無
輸入輸出樣例
輸入 #17 5 1 5 3 5 5 1 3 2 2 3 3 4 3輸出 #1
2 3 1 5
解題思路:
1.想到使用multimet
2.熟悉multiset的各個函式功能
參考程式碼:
#include<iostream>
#include<cmath>
#include<algorithm>
#include<set>
using namespace std;
multiset<int>q;
int n,t,x,order;
int main()
{
q.insert( -2147483647);
q.insert( 2147483647);
scanf("%d",&n);
while(n--)
{
scanf("%d%d",&t,&x);
if(t==1)
{
auto it=q.lower_bound(x);
order=0;
for(auto i=q.begin();i!=it;i++,order++);
printf("%d\n",order);
}
else if(t==2)
{
order=-1;
for(int i:q)
if(++order==x)
printf("%d\n",i);
}
else if(t==3)
{
auto it=q.lower_bound(x);
printf("%d\n",*--it);
}
else if(t==4)
{
printf("%d\n",*q.upper_bound(x));
}
else
{
q.insert(x);
}
}
return 0;
}