POJ-2104-线段树，bindary search

来源：互联网发布：淘宝小李84 编辑：程序博客网时间：2024/05/25 13:34

题目大意：给定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|1using 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;}

0 0