1、k位排序数组
遍历每个元素,与k位排序数组的第一位比较,如果大于排序数组的第一位,则替换之并重排序。注意如果排序数组不足k位,则先补足k位。
时间复杂度:O(n * (k的平方))
func main() {
fmt.Printf("val: %v ", slove([]int64{4,5,6,1,2,3}, 3))
}
func slove(arr []int64, k int64) (val int64) {
sortArr := make([]int64, 0)
for _, item := range arr {
if int64(len(sortArr)) < k {
sortArr = append(sortArr, item)
sort(&sortArr)
} else {
if item > sortArr[0] {
sortArr[0] = item
sort(&sortArr)
}
}
}
val = sortArr[0]
return
}
func sort(sortArr *[]int64) {
if sortArr == nil {
return
}
for i := 0; i < len(*sortArr)-1; i++ {
for j := 0; j < len(*sortArr)-1-i; j++ {
if (*sortArr)[j] > (*sortArr)[j+1] {
(*sortArr)[j], (*sortArr)[j+1] = (*sortArr)[j+1], (*sortArr)[j]
}
}
}
return
}
2、小顶堆
建一个k个元素的小顶堆,遍历数组的每个值如果大于堆顶则取代堆顶,对新堆顶进行向下调整。
时间复杂度:O(nlogk)
空间复杂度:k
(代码略)