NC88寻找第K大的数

题目描述

有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。

给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在。

测试样例：

[1,3,5,2,2],5,3

返回：2

题解：

利用快速排序的方法，找准第一个数的最终位置，下标为index。如果它的右侧数量正好是k-1个，那么它就是第k大的数，因为比它大的数最终都会在它的右边；如果右侧比k-1小，则要找的数在左边，递归求解左边第k-（右侧个数+1）

 

package com.wx.nowcoder;

public class NC88 {
    public int findKth(int[] a, int n, int K) {
        return find(a, 0, n - 1, K);
    }

    private int find(int[] a, int left, int right, int k) {
        int index = partion(a, left, right);
        if (right - index + 1 == k) {
            return a[index];
        }
        if (right - index + 1 < k) {
            return find(a, left, index - 1, k - right + index - 1);
        }
        return find(a, index + 1, right, k);
    }


    private int partion(int[] arr, int left, int right) {
        int key = arr[left];
        while (left < right) {
            while (left < right && arr[right] >= key) {
                right--;
            }
            arr[left] = arr[right];
            while (left < right && arr[left] <= key) {
                left++;
            }
            arr[right] = arr[left];
        }
        arr[left] = key;
        return left;
    }

    public static void main(String[] args) {
        NC88 nc = new NC88();
        int[] a = {4, 1, 3, 5, 2};
        int k = 3;
        int i = nc.findKth(a, a.length, k);
        System.out.println(i);
    }
}