hdu 4417 划分树

思路:二分枚举区间第k大。用划分树查找是否符合要求的高度。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#define Maxn 100010
#define lson(x) x<<1
#define rson(x) x<<1|1
#define mid ((tree[po].l+tree[po].r)>>1)
using namespace std;
int val[][Maxn],toLeft[][Maxn],sorted[Maxn];
struct Tree{
int l,r;
}tree[Maxn*];
void BuildTree(int l,int r,int dep,int po)
{
tree[po].l=l,tree[po].r=r;
if(l==r)
return ;
int same=mid-l+,i;
for(i=l;i<=r;i++)
if(val[dep][i]<sorted[mid])
same--;
int lpos=l,rpos=mid+;
for(i=l;i<=r;i++){
if(i==l)
toLeft[dep][i]=;
else
toLeft[dep][i]=toLeft[dep][i-];
if(val[dep][i]<sorted[mid])
toLeft[dep][i]++,val[dep+][lpos++]=val[dep][i];
else
if(val[dep][i]>sorted[mid])
val[dep+][rpos++]=val[dep][i];
else
if(same)
toLeft[dep][i]++,val[dep+][lpos++]=val[dep][i],same--;
else
val[dep+][rpos++]=val[dep][i];
}
BuildTree(l,mid,dep+,lson(po));
BuildTree(mid+,r,dep+,rson(po));
}
int query(int l,int r,int k,int dep,int po)
{
if(l==r)
return val[dep][l];
int vd,invd;
if(l==tree[po].l){
vd=toLeft[dep][r];
invd=;
}
else{
vd=toLeft[dep][r]-toLeft[dep][l-];
invd=toLeft[dep][l-];
}
if(vd>=k){
int newl=invd+tree[po].l;
int newr=invd+vd+tree[po].l-;
return query(newl,newr,k,dep+,lson(po));
}
else{
int ss=l-tree[po].l-invd;
int s=r-l+-vd;
int newl=mid+ss+;
int newr=mid+ss+s;
return query(newl,newr,k-vd,dep+,rson(po));
}
}
int solve(int s,int t,int x)
{
int i,j,l,r,Mid;
l=,r=t-s+;
while(l+<r){
Mid=(l+r)>>;
int temp=query(s,t,Mid,,);
if(temp>x)
r=Mid-;
else
l=Mid;
}
if(r>=)
if(query(s,t,r,,)<=x)
return r;
return l;
}
int main()
{
int n,m,i,j,l,r,Case=,t,x;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(i=;i<=n;i++){
scanf("%d",&val[][i]);
sorted[i]=val[][i];
}
sort(sorted+,sorted++n);
BuildTree(,n,,);
printf("Case %d:\n",++Case);
for(i=;i<=m;i++){
scanf("%d%d%d",&l,&r,&x);
l++,r++;
printf("%d\n",solve(l,r,x));
}
}
return ;
}
上一篇:Android自定义控件4--优酷菜单的菜单键及细节补充


下一篇:RobotFramework下的http接口自动化Get Response header 关键字的使用