本文主要是介绍无序数组中找第K个的数,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目分析:也就是从小往大排,第K小那个数。
方法1:排序(NlogN)
方法2:利用堆(NlogK)
首先将前K个元素构建最大堆,堆顶是前K个元素中第K小的元素。(这步复杂度KlogK)
遍历剩余元素(这步复杂度(N-K)logK)
如果新元素<堆顶 堆顶不可能是第K大元素 移除堆顶 将新元素插入堆
否则 新元素不可能是第K大的元素
返回堆顶就是结果
方法3:利用快排Partition O(N)
递归方法
int kth(Integer[]A, int left, int right, int k) {if(A == null || k < left || k > right) return Integer.MAX_VALUE;int pivot = Partition.partition(A, left, right);if(pivot == k) return A[k];else if(pivot < k) return kth(A, pivot + 1, right, k);else return kth(A, left, pivot - 1, k);
}
非递归方法
int kth(Integer[]A, int left, int right, int k) {if(A == null || k < left || k > right) return Integer.MAX_VALUE;while(true) {int pivot = Partition.partition(A, left, right);if(pivot == k) return A[k];else if(pivot < k) left = pivot + 1;else right = pivot - 1; }
}
复杂度计算:O(N + N/2 + N/4 +...+ 1) = O(2N) = O(N)
拓展题
两个无序数组找第K(先合并 再找第K)
两个有序数组找第K https://oj.leetcode.com/problems/median-of-two-sorted-arrays/
这篇关于无序数组中找第K个的数的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!