力扣 414. 第三大的数

题目

给你一个非空数组，返回此数组中 第三大的数 。如果不存在，则返回数组中最大的数。

示例 1：

输入：[3, 2, 1]
输出：1
解释：第三大的数是 1 。

示例 2：

输入：[1, 2]
输出：2
解释：第三大的数不存在, 所以返回最大的数 2 。

示例 3：

输入：[2, 2, 3, 1]
输出：1
解释：注意，要求返回第三大的数，是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数，它们都排第二。在所有不同数字中排第三大的数为 1 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/third-maximum-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：优先队列

Java实现

class Solution {
    public int thirdMax(int[] nums) {
        PriorityQueue<Integer> q = new PriorityQueue<>();
        Set<Integer> set = new HashSet<>();
        for (int num : nums) {
            if (!set.contains(num)) {
                if (q.size() < 3) {
                    q.offer(num);
                } else {
                    if (num > q.peek()) {
                        q.poll();
                        q.offer(num);
                    }
                }
                set.add(num);
            }
        }
        if (q.size() == 3) {
            return q.peek();
        }
        int res = 0;
        while (!q.isEmpty()) {
            res = q.poll();
        }
        return res;
    }
}

方法2

class Solution {
    public int thirdMax(int[] nums) {
        int first_max = Integer.MIN_VALUE;
        int second_max = Integer.MIN_VALUE;
        int third_max = Integer.MIN_VALUE;
        // HashSet不会添加重复的元素
        HashSet<Integer> new_nums = new HashSet<>();
        for (int i = 0; i < nums.length; i++) {
            new_nums.add(nums[i]);
        }
        // 迭代HashSet
        for (Integer integer : new_nums) {
            // 记录最大值
            if (integer >= first_max) {
                third_max = second_max;
                second_max = first_max;
                first_max = integer;
            }
            // 记录第二大值
            else if (integer >= second_max) {
                third_max = second_max;
                second_max = integer;
            }
            // 记录第三大值
            else if (integer >= third_max) {
                third_max = integer;
            }
        }
        // 如果没有第三大的数,返回最大值
        if (new_nums.size() < 3) {
            return first_max;
        }

        return third_max;
    }
}