HDU4006 求集合第K大数（线段树+离散化）

题意：

输入 I 表示在集合中插入一个数，Q表示询问当前集合中第k大数

思路：

1.树状数组加二分，可是这个算法是n(logn)^2的，对于1e6肯定TLE，

但是树状数组有一个求区间第k大的黑科技，可以自己搜一下。

2.由于本蒟篛希望能练练线段树，于是本题我用线段树

真正的题解！

首先嘛，我不确定加入集合的数的大小范围（题目似乎没给？？）

于是，果断先把修改和询问先离线下来再  离散一波！

接着，在线段树上单点修改，每次将离散后的序号传进去

最重要的来了，要在线段树上二分！

如果一个节点的右儿子对应的区间和大于k那么继续向右儿子走，否则向左儿子走。

二分详细步骤在代码中

/*
	最怕一生碌碌无为，还说平凡难能可贵。
*/
#include<bits/stdc++.h>
#define INF 1001000
using namespace std;
struct data{
	int l,r,sum;
}tr[INF*4];
int n,k,linum;
int pos[INF],lisan[INF];
char ch[INF];
inline void R(int &x)
{
	x=0;
	char c=getchar();
	int p=1;
	while(c<'0'||c>'9')
	{
		if(c=='-')
			p=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9')
	{
		x=(x<<1)+(x<<3)+(c^'0');
		c=getchar();
	}
	x*=p;
}
void update(int k){tr[k].sum=tr[k<<1].sum+tr[k<<1|1].sum;}
void build(int k,int s,int t)
{
	tr[k].l=s;tr[k].r=t;
	if(s==t)
	{
		tr[k].sum=0;
		return;
	}
	int mid=s+t>>1;
	build(k<<1,s,mid);
	build(k<<1|1,mid+1,t);
	update(k);
}
void modify(int k,int poss)
{
	if(tr[k].l==tr[k].r)
	{
		tr[k].sum++;
		return;
	}
	int mid=tr[k].l+tr[k].r>>1;
	if(poss<=mid)modify(k<<1,poss);
	else modify(k<<1|1,poss);
	update(k);
}
int query(int k,int q)
{
	if(tr[k].l==tr[k].r)return lisan[tr[k].l];
	if(tr[k<<1|1].sum>=q)return query(k<<1|1,q);
	else return query(k<<1,q-tr[k<<1|1].sum);//"q-tr[k<<1|1].sum"是因为如果要继续访问左儿子，那么原来的第q大数，在左儿子中应该是求第 （q - 右儿子中统计的数的总和 ） 
}
int getlisanpos(int x){return lower_bound(lisan+1,lisan+linum+1,x)-lisan;}	
int main()
{
	//freopen("in.in","r",stdin);
	while(cin>>n>>k)
	{
		int cnt=0;
		for(int i=1;i<=n;i++)
		{
			char s[2];
			scanf("%s",s);
			ch[i]=s[0];
			if(ch[i]=='I')
			{
				R(pos[i]);
				lisan[++cnt]=pos[i];
			}
		}
		sort(lisan+1,lisan+cnt+1);
		linum=unique(lisan+1,lisan+cnt+1)-lisan-1;
		build(1,1,n);
		for(int i=1;i<=n;i++)
		{
			if(ch[i]=='I')
				modify(1,getlisanpos(pos[i]));
			else
				cout<<query(1,k)<<endl;
		}
	}
}