poj2104[划分树问题]

这个题目的意思是给你一个区间，让你找到在某个给出的子区间当中第k大的数。

这是我的第一个划分树题目，因为之前做了好多线段树的题目了，看了看书上基本的代码，然后按照自己的思路写了出来。

按照这个题目的思路，得用到二分，把给出的元素x
排好序y
，对于一个给出的子区间，那么第k大的元素应该满足两点：

（1）在区间当中至少存在k个不超过第k大的元素

（2）在区间当中小于第k大的元素不到k个

从排好序列的区间当中取出mid,划分树当中每个节点保存的是一个区间内排好序的元素，当这个节点所在的区间完全包含在被给出的区间的时候，就找出小于等于y[mid]的元素的个数，每次查询完成以后得出小于等于y[mid]的元素的个数t,如果t>=k,就让right=mid，否则的话就让left=mid;这样二分可以很快找到满足条件的y[mid]，最后输出的y[mid]就是查找到的区间当中的第k大的数

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>

#define tag 1200
#define maxn 100010
using namespace std;
int x[maxn];
int y[maxn];
vector<int>ve[4*maxn];
void build(int now,int l,int r)
{
if (r==l)
{
ve[now].clear();
ve[now].push_back(x[l]);
return ;
}
int mid=(l+r)>>1;
build(2*now,l,mid);
build(2*now+1,mid+1,r);
ve[now].resize(r-l+1);//目的是申请足够容纳[l,r]内元素的内存单元
merge(ve[2*now].begin(),ve[2*now].end(),ve[2*now+1].begin(),ve[2*now+1].end(),ve[now].begin());//由下往上区间合并，合并的同时完成了排序
}
int query(int now,int l,int r,int tl,int tr,int val)//查询操作，关键地方
{
if (tl>r||tr<l)return 0;
if (tl<=l&&tr>=r)
{
int t=upper_bound(ve[now].begin(),ve[now].end(),val)-ve[now].begin();
return t;
}
else
{
int mid=(l+r)>>1;
int t1=query(2*now,l,mid,tl,tr,val);
int t2=query(2*now+1,mid+1,r,tl,tr,val);
return t1+t2;
}
}
int main()
{
int n,m;
int shu[5]={4,5,6,7,8};
while(cin>>n>>m)
{
for (int i=1;i<=n;i++)
{
scanf("%d",&x[i]);
y[i]=x[i];
}
sort(y+1,y+1+n);
build(1,1,n);
int l,r,val,mid;
int sum;
while(m--)
{
scanf("%d %d %d",&l,&r,&val);
int left=0,right=n;
while(right-left>1)//二分方法找到在[l,r]内的第k大的元素
{
mid=(left+right)>>1;
sum=y[mid];
int t=query(1,1,n,l,r,sum);
if(t>=val)
{
right=mid;
}
else
{
left=mid;
}
}
printf("%d\n",y[right]);
}
}
return 0;
}