现在的位置: 首页 > 综合 > 正文

POJ 2104 K-th Number

2013年10月27日 ⁄ 综合 ⁄ 共 1281字 ⁄ 字号 评论关闭
/*划分树基础题模版*/
#include<cstdio>
#include<algorithm>
#include<iostream>

using namespace std;

const int M=100005;

int tree[20][M],sorted[M];
int toLeft[20][M];

void build(int level,int left,int right){
if(left==right)return ;
int mid=(left+right)>>1;
int i;
int suppose;
suppose=mid-left+1;
for(i=left;i<=right;i++){
if(tree[level][i]<sorted[mid]){
suppose--;
}
}

int lpos=left,rpos=mid+1;
for(i=left;i<=right;i++){
if(i==left){
toLeft[level][i]=0;
}else{
toLeft[level][i]=toLeft[level][i-1];
}
if(tree[level][i]<sorted[mid]){
toLeft[level][i]++;
tree[level+1][lpos++]=tree[level][i];
}else if(tree[level][i]>sorted[mid]){
tree[level+1][rpos++]=tree[level][i];
}else{
if(suppose!=0){
suppose--;
toLeft[level][i]++;
tree[level+1][lpos++]=tree[level][i];
}else{
tree[level+1][rpos++]=tree[level][i];
}
}
}
build(level+1,left,mid);
build(level+1,mid+1,right);
}

int query(int level,int left,int right,int qleft,int qright,int k){
if( qleft==qright)
return tree[level][qleft];
int s;
int ss;
int mid=(left+right)>>1;
if(left==qleft){
s=0;
ss=toLeft[level][qright];
}else{
s=toLeft[level][qleft-1];
ss=toLeft[level][qright]-s;
}
int newl,newr;
if(k<=ss){
newl=left+s;
newr=left+s+ss-1;
return query(level+1,left,mid,newl,newr,k);
}else{
newl=mid-left+1+qleft-s;
newr=mid-left+1+qright-s-ss;
return query(level+1,mid+1,right,newl, newr,k - ss);
}
}

int main(){
int n,m;
while(~scanf("%d%d",&n,&m))

抱歉!评论已关闭.