HDU4006 求集合第K大数(线段树+离散化)

题意:

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

思路:

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

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

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


真正的题解!

首先嘛,我不确定加入集合的数的大小范围(题目似乎没给??)

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

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

最重要的来了,要在线段树上二分!

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

二分详细步骤在代码中


/*
	最怕一生碌碌无为,还说平凡难能可贵。
*/
#include
#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<



你可能感兴趣的:(HDU4006 求集合第K大数(线段树+离散化))