树状数组求第k大

辣鸡南瓜并不能在考场上打出平衡树,就只能退而求其次,离线用树状数组来解决啦

板子题传送门:https://www.luogu.org/problemnew/show/P3369

code:

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <ctime>
#define maxn 105000
using namespace std;

int n,tot,sz;
int opt[maxn],val[maxn],num[maxn],c[maxn];

int read()
{
	int xx=0,kk=1;char ch=' ';
	while(!isdigit(ch)){ch=getchar();if(ch=='-')kk=-1;}
	while(isdigit(ch)){xx=xx*10+ch-'0';ch=getchar();}
	return kk*xx;
}

int pos(int x)
{
	return lower_bound(num+1,num+1+sz,x)-num;
}

int lowbit(int x)
{
	return x&-x;
}

void add(int x,int k)
{
	for(x;x<=sz;x+=lowbit(x)) c[x]+=k;
}

int query(int x)
{
	int tmp=0;
	for(x;x;x-=lowbit(x)) tmp+=c[x];
	return tmp;
}

int kth(int x)
{
	int rs=0;
	for(int i=17;i>=0;--i)
	{
		rs+=(1<<i);
		if(rs>sz||c[rs]>=x) rs-=(1<<i);
		else x-=c[rs];
	}
	return num[rs+1];
}

int main()
{
	n=read();
	for(int i=1;i<=n;++i)
	{
		opt[i]=read(),val[i]=read();
		if(opt[i]!=4) num[++tot]=val[i];
	}
	sort(num+1,num+1+tot);
	sz=unique(num+1,num+1+tot)-num-1;
	for(int i=1;i<=n;++i)
	{
		int x=val[i];
		switch(opt[i])
		{
			case 1:add(pos(x),1);break;
			case 2:add(pos(x),-1);break;
			case 3:printf("%d\n",query(pos(x)-1)+1);break;
			case 4:printf("%d\n",kth(x));break;
			case 5:printf("%d\n",kth(query(pos(x)-1)));break;
			case 6:printf("%d\n",kth(query(pos(x))+1));break;
		}
	}
	return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值