教育改变生活

标题: Java编程-寻找第K大 [打印本页]

作者: 一秉 时间: 2020-8-31 16:55
标题: Java编程-寻找第K大
题目描述：
有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在。
测试样例：
输入：[1,3,5,2,2],5,3
输出：2
解析思路：
用快排的思想：例如找49个元素里面第24大的元素，那么按如下步骤：
1.进行一次快排（将大的元素放在前半段，小的元素放在后半段）,假设得到的中轴为m
2.判断m + 1 == k ，如果成立，直接输出a[m]，（因为前半段有k -1个大于am]的元素，故a[p]为第K大的元素）
3.如果 m + 1 > k，则第k大的元素在前半段，此时更新right = m - 1，继续进行步骤1
4.如果m + 1 <  k，则第k大的元素在后半段，此时更新left= m + 1，继续步骤1.
由于常规快排要得到整体有序的数组，而此方法每次可以去掉“一半”的元素，故实际的复杂度不是o(nlgn), 而是o(n)。
代码：
import java.util.*;

public class Finder {
   public int findKth(int[] a, int n, int K) {
      return  help(0,n-1,a,K);
}

private int help(int left, int right, int[] a, int k) {
      int m=partation(left,right,a);
      if(m+1 > k){
         return  help(left,m-1,a,k);
      }else if(m+1<k){
         return help(m+1,right,a,k);
      }else {
         return a[m];
      }
}

private int partation(int left, int right, int[] a) {
      int pivotValue =a[left];
      while (left<right){
         while (left<right && a[right]<=pivotValue){
            right--;
         }
         a[left]=a[right];
         while (left<right && a[left]>=pivotValue){
            left++;
         }
         a[right]=a[left];
      }
      a[left]=pivotValue;
      return left;
}
}

欢迎光临教育改变生活 (http://bbs.goldoar.com/)