POJ-2104-线段树,bindary search

题目大意:给定n个数字和m次询问,问区间[l,r]上第k大的数字是多少;

题目解析:线段树,每个节点保存的是一个序列,所以pushup的时候需要用merge,归并排序;如果是第k大的,那么小于等于那个数字的就有k个,所以可以二分;

AC代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int inf=0x3fffffff;
const int maxn=100010;
int a[maxn];
vector<int>vec[maxn<<2];
void pushup(int rt)
{
	merge(vec[rt<<1].begin(),vec[rt<<1].end(),vec[rt<<1|1].begin(),vec[rt<<1|1].end(),vec[rt].begin());	
}
void build(int l,int r,int rt)
{
	if(l==r)
	{
		int val;
		scanf("%d",&val);
		vec[rt].push_back(val);
		a[l]=val;
		return ;
	}	
	int m=(l+r)>>1;
	build(lson);
	build(rson);
	vec[rt].resize(r-l+1);
	pushup(rt);
}
int query(int le,int ri,int x,int l,int r,int rt)
{
	if(ri<l||le>r)
	{
		return 0;
	}
	if(le<=l&&ri>=r)
	{
		return upper_bound(vec[rt].begin(),vec[rt].end(),x)-vec[rt].begin();
	}
	int m=(l+r)>>1;
	return query(le,ri,x,lson)+query(le,ri,x,rson); 
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	build(1,n,1);
	sort(a+1,a+n+1);
	while(m--)
	{
		int ll,rr,x;
		scanf("%d%d%d",&ll,&rr,&x);
		int l=1,r=n,ans;
		while(l<=r)
		{
			int m=(l+r)>>1;
			int num=query(ll,rr,a[m],1,n,1);
			if(num>=x)
			{
				r=m-1;
				ans=a[m];
			}
			else 
			{
				l=m+1;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值