Hiho : 二分·二分查找之k小数

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

在上一回里我们知道Nettle在玩《艦これ》，Nettle的镇守府有很多船位，但船位再多也是有限的。Nettle通过捞船又出了一艘稀有的船，但是已有的N(1≤N≤1,000,000)个船位都已经有船了。所以Nettle不得不把其中一艘船拆掉来让位给新的船。Nettle思考了很久，决定随机选择一个k，然后拆掉稀有度第k小的船。 已知每一艘船都有自己的稀有度，Nettle现在把所有船的稀有度值告诉你，希望你能帮他找出目标船。

提示：非有序数组的二分查找之二

输入

第1行：2个整数N,k。N表示数组长度，
第2行：N个整数，表示a[1..N]，保证不会出现重复的数，1≤a[i]≤2,000,000,000。

输出

第1行：一个整数t，表示t在数组中是第k小的数，若K不在数组中，输出-1。

样例输入

10 4
1732 4176 2602 6176 1303 6207 3125 1 1011 6600

样例输出

1732

代码：

import java.util.Arrays;
import java.util.Scanner;

public class Main {
public static void main(String [] argv){

Scanner in = new Scanner(System.in);
int N = in.nextInt();
int M = in.nextInt()-1;
int [] s = new int
;
for (int i=0; i<N; i++){
s[i]=in.nextInt();
}
in.close();
if(M+1>N||M<0)
System.out.println(-1);
else
KP(M,s);

}

static    public void KP(int m, int [] h){
int s[] =h;
int k = s[0];
int first = 0;
int last = s.length-1;
while(true){
while(first<last){
while(first<last&&s[last]>=k)
--last;
s[first]=s[last];
while(first<last&&s[first]<=k)
++first;
s[last]=s[first];
if(first==last)
s[first]=k;
}
for(int t = 0 ; t< s.length; t++){
//System.out.print(s[t]);
}
//System.out.println("first:"+first+"m:"+m+"last:"+last);
if(first<m){
int[] temp = new int[s.length-1-first];
for(int j = 0; j<s.length-1-first; j++){
temp[j]=s[j+first+1];
}
s = temp;
m = m-first-1;
first = 0;
last = s.length-1;
}
else{
if(first==m){
System.out.println(k);
return ;
}

else{
int[] temp = Arrays.copyOf(s, first);
s=temp;
first = 0;
last = s.length-1;
}

}
if(s.length>=1)
k=s[0];
}

}

}

通过效果：