在未排序的数组中找到最大的K个数字

时间:2019-03-17 00:58:05

标签: java algorithm partition quickselect

我正在尝试查找排序数组中最大的K个数字。

例如: 输入-> [5,12,45,32,9,20,15] 输出-> K = 3,[45,32,20]

到目前为止,我编写的代码返回最大的K元素,但是它需要返回最大的K个数字。任何帮助将不胜感激。

public static int max_Numbers(int [] p, int K, int firstNum, int lastNum)
    {
        int pivot = partitionArr(p, firstNum, lastNum);
        int m = p.length - K;
        if (m == pivot)
        {
            return p[pivot];
        }
        if(m > pivot)
        {
            return max_Numbers(p, K, pivot + 1, lastNum);

        }
        else
        {
            return max_Numbers(p, K, firstNum, pivot - 1);
        }
    }

2 个答案:

答案 0 :(得分:0)

使用排序后的数组,

for(int i=array.length-1; i>=0 && array.length-1 - i < K; i--) System.out.println(array[i]));

答案 1 :(得分:0)

您正在使用的枢轴和分区的一个属性是,在每个分区步骤之后,可以确保枢轴之前的所有元素都小于或等于枢轴,并且枢轴之后的所有元素都更大。因此,在找到第K个枢轴之后,该数组将在其后具有最大的K。