首页 > 代码库 > 找一个数组中第K大的数

找一个数组中第K大的数

快排思想,选取序列的一个key进行划分,小于key的分在左边,大于key的在右边,key的位置-low+1就是最后一个元素是key的序列中元素的数量,当元素数量大于K时,就在左半部分递归找,等于时 arr[key]就是第K

大的元素,小于K时,在右边递归找第k-num大的元素

/**
 * 文件名:FindK.java
 * 时间:2014年11月7日上午11:23:43
 * 作者:修维康
 */
package chapter7;

/**
 * 类名:FindK 说明:找到一个数组中第K大的元素
 */
public class FindK {

	public static int partition(int[] a, int low, int high) {
		int key = a[low];
		while (low < high) {
			while (low < high && a[high] >= key)
				high--;
			a[low] = a[high];
			while (low < high && a[low] <= key)
				low++;
			a[high] = a[low];
		}
		a[low] = key;
		return low;
	}

	public static int quickSelect(int[] a, int low, int high, int k) {

		if (low == high)
			return a[low];
		int keyPos = partition(a, low, high);
		
		int num = keyPos - low + 1;
		if (num == k)
			return a[keyPos];
		else if (k < num)
			return quickSelect(a, low, keyPos - 1, k);
		else
			return quickSelect(a, keyPos + 1, high, k - num);

	}

	/**
	 * 方法名:main 说明:测试
	 */
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		int[] a = new int[] {1,2,3,4,5,6,7,8,9,10};
		System.out.println(quickSelect(a, 0, a.length - 1,5));

	}
}


找一个数组中第K大的数