寻找第K大 --- 牛客网

题目链接

https://www.nowcoder.com/questionTerminal/e016ad9b7f0b45048c58a9f27ba618bf

描述

链接：https://www.nowcoder.com/questionTerminal/e016ad9b7f0b45048c58a9f27ba618bf
来源：牛客网

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

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

示例1
输入
[1,3,5,2,2],5,3
输出
2

代码

利用快排思想，每次都找到当前元素的位置，符合要求就直接返回，不符合继续寻找

import java.util.*;

public class Finder {
    public int findKth(int[] a, int n, int K) {
        // write code here
        return findKth(a, 0, n - 1, n - K);
    }
    
    private int findKth(int[] a, int low, int high, int K) {
        int part = partion(a, low, high);
        
        if (part == K) {
            return a[part];
        } else if (part < K) {
            return findKth(a, part + 1, high, K);
        } else {
            return findKth(a, low, part - 1, K);
        }
    }
    
    private int partion(int[] a, int low, int high) {
        int temp = a[low];
        
        while (low < high) {
            while (low < high && a[high] > temp) {
                high--;
            }
            a[low] = a[high];
            while (low < high && a[low] <= temp) {
                low++;
            }
            a[high] = a[low];
        }
        
        a[low] = temp;
        return low;
    }
}