第k大的数 java_一道算法题：第K大的数
給一個無序的包涵n個元素的數組，找出其中第k大的數(n > k)。
初看到這個題的時候，作為一個寫了一段時間java的人，立刻能想到的一種解法就是：
class Solution {
public int kthLargestElement(int k, int[] nums) {
Arrays.sort(nums);
return nums[nums.length - k];
}
};
時間復雜度時NlgN, 空間復雜度時O(1).
看起來貌似不錯，但既然這道題目被廣泛地討論，就肯定還有其他解法，比如說：
class Solution {
public int kthLargestElement(int k, int[] nums) {
// write your code here
PriorityQueue pq = new PriorityQueue<>();
for(int val : nums) {
pq.add(val);
if(pq.size() > k) {
pq.poll();
}
}
return pq.peek();
}
};
當然建堆需要額外的空間，而且時間復雜度上并沒有多少提升，所以不能算什么優化。
更好的一種解法是利用快速排序的劃分思想，這種解法是一種O(n)的解法
class Solution {
public int kthLargestElement(int k, int[] nums) {
//打亂數組避免最壞情況
shuffle(nums);
k = nums.length - k;
int lo = 0, hi = nums.length - 1;
while(lo < hi){
int j = partition(nums, lo, hi);
if(j < k){
lo = j + 1;
}else if(j > k){
hi = j - 1;
}else{
break;
}
}
return nums[k];
}
private int partition(int[] nums, int lo, int hi){
int i = lo;
int j = hi + 1;
while(true){
while(i < hi && nums[++i] < nums[lo]);
while(j > lo && nums[lo] < nums[--j]);
if(j <= i) break;
exch(nums, i, j);
}
exch(nums, lo, j);
return j;
}
private void exch(int[] nums, int i, int j){
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
private void shuffle(int a[]) {
Random random = new Random();
for(int ind = 1; ind < a.length; ind++) {
int r = random.nextInt(ind + 1);
exch(a, ind, r);
}
}
};
總結
以上是生活随笔為你收集整理的第k大的数 java_一道算法题：第K大的数的全部內容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： mysql中sql命令大小写吗_sql语
- 下一篇： java把一个list_java中将一个