leetCode177|第三大的数-CSDN博客

本文链接：https://blog.csdn.net/qq_35006660/article/details/112792124

一，第三大的数

1，问题简述

给定一个非空数组，返回此数组中第三大的数。如果不存在，则返回数组中最大的数。要求算法时间复杂度必须是O(n)。

2，示例描述

示例 1:

输入: [3, 2, 1]

输出: 1

解释: 第三大的数是 1.
示例 2:

输入: [1, 2]

输出: 2

解释: 第三大的数不存在, 所以返回最大的数 2 .
示例 3:

输入: [2, 2, 3, 1]

输出: 1

解释: 注意，要求返回第三大的数，是指第三大且唯一出现的数。
存在两个值为2的数，它们都排第二。

3，题解思路

使用treeSet数据结构进行解决

4，题解程序

import java.util.TreeSet;

public  class ThirdMaxTest {
    public static void main(String[] args) {
        int[] nums = {2, 2, 3, 1};
        thirdMax(nums);
    }

    public static int thirdMax(int[] nums) {
        TreeSet<Integer> treeSet = new TreeSet<>();
        for (int i = 0; i < nums.length; i++) {
            treeSet.add(nums[i]);
            if (treeSet.size() > 3) {
                treeSet.pollFirst();
            }
        }
        return treeSet.size() > 2 ? treeSet.pollFirst() : treeSet.pollLast();
    }
}