414. 第三大的数

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

示例 1：

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

• 把元素都插入TreeSet里面,他会自动升序排列
• 在插入过程中,一直维护一个长度为3的,如果大于3,那么删除最小的那个
• 插入完毕
• 没有第三个 就返回最后一个(最大值) ,否则返回第一个(一直维护第一个是倒数第三大的)

class Solution {
    public int thirdMax(int[] nums) {
        TreeSet<Integer> set=new TreeSet();
        for(int i:nums){
            set.add(i);
            if(set.size()>3){
                 set.pollFirst();
            }
        }
        if(set.size()<3){
            return set.pollLast();
        }else{
            return set.pollFirst();
        }

    }
}