1. 程式人生 > 其它 >【深基16.例7】普通二叉樹(簡化版)

【深基16.例7】普通二叉樹(簡化版)

題目:【深基16.例7】普通二叉樹(簡化版)

題目連結:https://www.luogu.com.cn/problem/T237469?contestId=68358

題目描述

您需要寫一種資料結構,來維護一些數( 都是 10^9109 以內的數字)的集合,最開始時集合是空的。其中需要提供以下操作,操作次數 qq 不超過 10^4104:

  1. 查詢 xx 數的排名(排名定義為比當前數小的數的個數 +1+1。若有多個相同的數,應輸出最小的排名)。
  2. 查詢排名為 xx 的數。
  3. 求 xx 的前驅(前驅定義為小於 xx,且最大的數)。若未找到則輸出 -21474836472147483647。
  4. 求 xx 的後繼(後繼定義為大於 xx,且最小的數)。若未找到則輸出 21474836472147483647。
  5. 插入一個數 xx。

輸入格式

輸出格式

輸入輸出樣例

輸入 #1
7
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;
}