题意:设计一种数据结构,该数据结构有以下三种操作:
1:0 a,添加一个数字a;
2:1 a,删除一个数字a,若该数字不存在,输出"No Elment!";
3:2 a k,输出比a大的第k个数字,若不存在,输出"Not Find!"。
分析:典型的用树状数组求第k大数字,对应操作:
1 :add(a,1);
2:add(a,-1);
3:只需先求出小于等于a的数字有多少个,若有x个,则问题就变成了求第x+k的数字是多少,二分找出就行了。
代码如下:
#include <cstdio>
#include <stack>
#include <set>
#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include <list>
#include <functional>
#include <cstring>
#include <algorithm>
#include <cctype>
#include <string>
#include <map>
#include <iomanip>
#include <cmath>
#define LL long long
#define ULL unsigned long long
#define SZ(x) (int)x.size()
#define MP(a, b) make_pair(a, b)
#define MS(arr, num) memset(arr, num, sizeof(arr))
#define PB push_back
#define F first
#define S second
#define ROP freopen("input.txt", "r", stdin);
#define MID(a, b) (a + ((b - a) >> 1))
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define lrt rt << 1
#define rrt rt << 1|1
#define root 1,n,1
#define BitCount(x) __builtin_popcount(x)
#define BitCountll(x) __builtin_popcountll(x)
#define LeftPos(x) 32 - __builtin_clz(x) - 1
#define LeftPosll(x) 64 - __builtin_clzll(x) - 1
const double PI = acos(-1.0);
const int INF = 1e7;
using namespace std;
const double eps = 1e-5;
const int MAXN = 300 + 10;
const int MOD = 1000007;
const double M=1e-8;
const int Max=100010;
typedef pair<int, int> pii;
typedef pair<int, string> pis;
const int d[4][2]={{0,1},{0,-1},{-1,0},{1,0}};
int n,m,c[Max],mx,vis[Max] ;
int Lowbit(int t)
{
return (t & ( -t ));
}
int sum(int end)
{
int ans = 0;
while(end > 0)
{
ans += c[end];
end -= Lowbit(end);
}
return ans;
}
void add(int pos,int num)
{
while(pos <= Max)
{
c[pos] += num;
pos += Lowbit(pos);
}
}
int bin(int x)
{
if (sum(Max-1)<x) return -1;
int l=1,r=Max;
while(l<r) {
int mid=MID(l,r);
int t=sum(mid);
if (x<=t) {
r=mid;
}
else if (x>t) l=mid+1;
}
return l;
}
int main()
{
int i,j;
while(~scanf("%d",&n))
{
MS(c,0);
MS(vis,0);
for (i=0;i<n;i++) {
int p,a,k;
scanf("%d%d",&p,&a);
if (p==0) {
add(a,1);
vis[a]++;
}
else if (p==1) {
if (vis[a]) {
add(a,-1);
vis[a]--;
}
else puts("No Elment!");
}
else {
scanf("%d",&k);
int ans=bin(sum(a)+k);
if (ans!=-1) printf("%d\n",ans);
else puts("Not Find!");
}
}
}
}