寻找第k大数字

最新推荐文章于 2021-05-25 08:25:08 发布

测开小白

最新推荐文章于 2021-05-25 08:25:08 发布

阅读量123

点赞数

本文链接：https://blog.csdn.net/weixin_43688527/article/details/103100599

版权

该博客介绍了如何使用快速排序的策略，在给定整数数组中找到第K大的数。通过分治算法，将大于关键值的元素置于前部，小于关键值的元素置于后部，然后根据中间位置的下标与K的关系进行递归查找，以确定第K大的元素。给出的测试样例为[1,3,5,2,2]，要求找到第3大的数，返回结果为2。" 23296041,2091653,C语言实现蛇形填数算法,"['C语言', '数组操作', '算法实现']

摘要由CSDN通过智能技术生成

题目

有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。

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

测试样例：
[1,3,5,2,2],5,3
返回：2

思想

利用快排思想，利用分治算法，把大于key的数字放在前面，把小于key的数字放后面，返回中间那个数的下标result,如果result-left==k-1,说明前面有k-1个比它大的数，如果result-left>k-1,就要从它的前面找,right=result-1，如果result-left<k-1,就要从后面找，left=result+1,这时不是和k比较，而是和k-1-(result-left)比较，因为可以一次性排除前面result-left+1个元素。

import java.util.*;

public class Finder {
    public int findKth(int[] a, int n, int K) {
        // write code here
    return find(a,0,n-1,K);
    }

    private int find(int[] a, int l, int r, int k) {
        int result=preorder(a,l,r);
        if(result-l+1==k){
            return a[result];
        }else if(result-l>k-1){
           return  find(a,l,result-1,k);
        }else
           return  find(a,result+1,r,k+l-result-1);

    }

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