LeetCode-414-第三大的數

雄獅虎豹發表於2021-10-11
LeetCode

第三大的數

題目描述：給你一個非空陣列，返回此陣列中 第三大的數 。如果不存在，則返回陣列中最大的數。

示例說明請見LeetCode官網。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/probl...
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

解法一：使用優先佇列

首先，初始化一個優先佇列PriorityQueue為queue，然後遍歷nums中的元素，遍歷過程如下：

  • 如果當前元素已經在queue中，則跳過，否則將當前元素新增到queue中；
  • 如果queue的元素數量大於3，則從queue中移除一個元素（溢位的元素為最小的）。

遍歷完成後，如果queue只有3個元素，則取出queue中的第一個元素並返回，即為nums中第三大的元素；如果queue沒有3個元素，則取出最後一個元素返回即為nums中的最大元素並返回。

import java.util.PriorityQueue;

/**
 * @Author: ck
 * @Date: 2021/9/29 8:08 下午
 */
public class LeetCode_414 {
    /**
     * 使用優先佇列
     *
     * @param nums
     * @return
     */
    public static int thirdMax(int[] nums) {
        PriorityQueue<Integer> queue = new PriorityQueue(3);
        for (int num : nums) {
            // 重複的元素過濾掉
            if (!queue.contains(num)) {
                queue.add(num);
            }
            // 因為獲取的是第三大的元素，所以當queue中的元素數量超過3時，需要將當前最小的元素取出
            if (queue.size() > 3) {
                queue.poll();
            }
        }
        if (queue.size() == 3) {
            // 當queue只有3個元素時，取最小的即為第三大的元素
            return queue.peek();
        } else {
            // 當queue的元素數量不夠3個時，遍歷queue取最大的一個
            int max = -1;
            while (!queue.isEmpty()) {
                max = queue.poll();
            }
            return max;
        }
    }

    public static void main(String[] args) {
        int[] nums = new int[]{1, 2, 2, 5, 3, 5};
        System.out.println(thirdMax(nums));
    }
}
【每日寄語】 生命的意義在於燃燒自己的同時能否照亮自己。

相關文章