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[N];
        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];
        }
        
    }


}

 

通过效果：