算法 寻找第k大数-(快速排序+双指针)

牛客网: BM47

题目: 数组第K大的数

思路: 见最小的k个数，将num[right] > pivot的元素左调，最终返回num[k-1]

代码:

// go

package main
// import "fmt"

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param a int整型一维数组 
 * @param n int整型 
 * @param K int整型 
 * @return int整型
*/
func findKth( input []int ,  n int ,  k int ) int {
    // write code here
    if len(input) == 0 || len(input) < k || k == 0 {
        return -1
    }
    low := 0
    high := len(input) - 1
    for low < high {
        left := low
        right := low
        pivot := input[high]
        for right < high {
            if input[right] > pivot {
                input[left], input[right] = input[right], input[left]
                left++
                right++
            } else {
                right++
            }
        }
        input[left], input[high] = input[high], input[left]
        if left == k - 1 {
            break
        } else if left > k - 1 {
            high = left - 1
        } else {
            low = left + 1
        }
    }
    return input[k-1]
}