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

题目1 : 二分·二分查找之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

利用快速函数的divide函数，每一次divide都会返回一个基准，设该基准为d

那么对于数组arr中  arr[0...k -1] < arr[d] and arr[d] < arr[d+1, ...n]

那么对于k,如果k小于d的话，相当于 在 0 - d - 1中寻找第k小的数

如果k 大于d的话，相当于在d + 1到n中寻找第k - d大的数

代码实现十分简单，如下

不过我发现对其排序后输出坐标k的数字也不会超时，想想也是醉了

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.Arrays;
import java.util.Scanner;


public class Main {
	public static void main(String[] args) throws Exception{
		int n, k;
		InputStream ins; 
		ins = System.in;
//		ins = new FileInputStream(new File("."));
		Scanner cin = new Scanner(System.in);
		n = cin.nextInt();
		k = cin.nextInt();
		int[] arr = new int[n];
		for(int i = 0; i < arr.length; ++i){
			arr[i] = cin.nextInt();
		}
//		Arrays.sort(arr);
		int result = -1;
		int left = 0, right = arr.length - 1;
		if(k > arr.length){
		    System.out.println(result);
		    return;
		}
		--k;
		do{
			result = searchKthNumber(arr, left, right);
			if(k < result){
			    right = result - 1;
			}
			else if(k > result){
			    left = result + 1;
			}else break;
		}while(true);
		System.out.println(arr[result]);
	}
	
	public static int searchKthNumber(int[] arr, int left, int right){
		int base = arr[left];
		while(left < right){
			while(left < right && arr[right] > base) -- right;
			arr[left] = arr[right];
			while(left < right && arr[left] < base) ++left;
			arr[right] = arr[left];
		}
		arr[left] = base;
		return left;
	}
}