LeetCode71|数组中第K个最大元素

时间:2022-07-26
本文章向大家介绍LeetCode71|数组中第K个最大元素,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

1,问题简述

在未排序的数组中找到第 k 个最大的元素。

请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

2,示例

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:

你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。

3,题解思路

排序,堆

4,题解程序

 
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class FindKthLargestTest2 {
    public static void main(String[] args) {
        int[] nums = {3, 2, 1, 5, 6, 4};
        int k = 2;
        int kthLargest = findKthLargest2(nums, k);
        System.out.println("kthLargest = " + kthLargest);
    }

    public static int findKthLargest(int[] nums, int k) {
        if (nums == null || nums.length == 0 || nums.length < k) {
            return -1;
        }
        Arrays.sort(nums);
        List<Integer> list = new ArrayList<>();
        for (int j : nums) {
            list.add(j);
        }
        Collections.sort(list, Comparator.reverseOrder());

        return list.get(k - 1);
    }

    public static int findKthLargest2(int[] nums, int k) {
        if (nums == null || nums.length == 0 || nums.length < k) {
            return 0;
        }

        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(k);
        for (Integer num : nums) {
            if (priorityQueue.size() < k || num > priorityQueue.peek()) {
                priorityQueue.offer(num);

            }
            if (priorityQueue.size() > k) {
                priorityQueue.poll();
            }
        }

        return priorityQueue.peek();
    }
}

5,题解程序图片版

6,总结

怎么样?最近的文章排版是不是比以前好太多了,人嘛,总是一步一步在成长,从最开始的文章到现在的文章,自己也在一步一步去做,看在眼里,记在心里。