NC88-寻找第K大的数
生活随笔
收集整理的這篇文章主要介紹了
NC88-寻找第K大的数
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
描述
有一個整數數組,請你根據快速排序的思路,找出數組中第K 大的數。
給定一個整數數組?,同時給定它的大小n和要找的K ,請返回第 K大的數(包括重復的元素,不用去重),保證答案存在。
要求時間復雜度 0(n)
示例1
輸入:
[1,3,5,2,2],5,3返回值:
2示例2
輸入:
[10,10,9,9,8,7,5,6,4,3,4,2],12,3返回值:
9說明:
去重后的第3大是8,但本題要求包含重復的元素,不用去重,所以輸出9java代碼:
import java.util.*;public class Solution {public int quicksort(int[] a,int left,int right,int K){int l=left,r=right;int key=a[left];while(r>l){while(a[r]>=key&&r>l){r--;}a[l]=a[r];while(r>l&&a[l]<=key){l++;}a[r]=a[l];}a[l]=key;//l為基準軸,l+1表示是數組中第l+1小的數if(l+1==K) {return a[l];}else if(l+1>K)return quicksort(a,left,l-1,K);else {return quicksort(a,l+1,right,K);}}public int findKth(int[] a, int n, int K) {if(a.length<K) return 0; //注意題目是尋找第K大的數,并不是第K小,排序的順序從小到大的順序,是第K小//所以要轉換 K1=n-K+1 ,轉換成第K小int value=quicksort(a,0,a.length-1,n-K+1);return value; } }java 題解2:
public class DemoTest {public static void main(String[] args) {//[10,10,9,9,8,7,5,6,4,3,4,2],12,3int[] a={10,10,9,9,8,7,5,6,4,3,4,2};int num=a.length;int K=3;int value=findKth(a, num, K);System.out.println(value);}public static int quicksort(int[] a,int left,int right,int K){int l=left,r=right;int key=a[left];while(r>l){while(a[r]>=key&&r>l){r--;}a[l]=a[r];while(r>l&&a[l]<=key){l++;}a[r]=a[l];}a[l]=key;int min=l-left; //min表示比基準軸l小的數有min個數,min+1代表整個數組中第min+1小的數if(min+1==K) { return a[l];}else if(min+1>K) //基準軸落在了目標值k的右邊,k值不變return quicksort(a,left,l-1,K);else {//基準軸落在了目標值k的左邊,k值改變,k=k-min-1,表示右半邊中相對于距離l+1的位置第k-min-1小的數return quicksort(a,l+1,right,K-min-1);}}public static int findKth(int[] a, int n, int K) {if(a.length<K) return 0; //注意題目是尋找第K大的數,并不是第K小,排序的順序從小到大的順序,是第K小//所以要轉換 K1=n-K+1 ,轉換成第K小int value=quicksort(a,0,a.length-1,a.length-K+1);return value; }}總結
以上是生活随笔為你收集整理的NC88-寻找第K大的数的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: NC119-最小的k个数
- 下一篇: NC61-两数之和