LeetCode【寻找第3大的数字】

题目描述:

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

测试用例:

输入：[3, 2, 1]
输出：1
解释：第三大的数是 1 
输入：[2, 2, 3, 1]
输出：1
解释：注意，要求返回第三大的数，是指第三大且唯一出现的数。
存在两个值为2的数，它们都排第二。

思路:

- 判断数组的合法性,数组长度为0或者数组为空,直接退出
- 将数组中的元素放入TreeSet
- 放入元素的时候判断set的size是否大于3,如果大于3就将首元素删除
- 插入完毕后,如果size大于等于3就返回首元素,否则返回尾巴元素

代码如下:

public class Solution {
    public int thirdMax(int[] nums) {
        if (nums.length == 0 || nums == null){
            System.exit(-1);
        }
        TreeSet<Integer> set = new TreeSet<>();
        for (Integer i : nums){
            set.add(i);
            if (set.size() > 3){
                set.remove(set.first());
            }
        }
        return set.size()>=3 ? set.first() : set.last();
    }
}