寻找第K大

题目描述:
有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。

给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数,保证答案存在。

思路分析:
这题是用快排的思想:例如找49个元素里面第24大的元素,那么按如下步骤: 1.进行一次快排(将大的元素放在前半段,小的元素放在后半段),假设得到的中轴为par 2.判断 par - begin + 1 == k ,如果成立,直接输出a[par],(因为前半段有k - 1个大于a[par]的元素,故a[par]为第K大的元素) 3.如果 par - begin + 1 > k, 则第k大的元
素在前半段,此时更新end = par - 1,继续进行步骤1 4.如果par - begin + 1 < k, 则第k大的元素在后半段, 此时
更新begin = par + 1, 且 k = k - (par - begin + 1),继续步骤1. 由于常规快排要得到整体有序的数组,而此方法每次可
以去掉“一半”的元素,故实际的复杂度不是o(nlgn), 而是o(n)。

import java.util.*;

public class Finder {
    public int findKth(int[] a, int n, int K) {
        // write code here
        return findKth(a,0,n-1,K);
    }
    public int findKth(int[] a,int begin,int end,int K){
        int par=partion(a,begin,end);
        if(K==par-begin+1){
            return a[par];
        }else if(K>(par-begin+1)){
            return findKth(a,par+1,end,K-par+begin-1);
        }else{
            return findKth(a,begin,par-1,K);
        }
    }
    public int partion(int[] a,int begin,int end){
        int key=a[begin];
        while(begin<end){
            while(end>begin&&a[end]<key){
                end--;
            }
            a[begin]=a[end];
            while(begin<end&&a[begin]>key){
                begin++;
            }
            a[end]=a[begin];
        }
        a[begin]=key;
        return begin;
    }
}
发布了172 篇原创文章 · 获赞 13 · 访问量 4682

猜你喜欢

转载自blog.csdn.net/qq_42174669/article/details/105246740
今日推荐