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();
    }
}
上一篇:Java-集合(set,map)


下一篇:Java 8 新特性 用 Collectors 对 List 去重