题目描述:
给你一个非空数组,返回此数组中第三大的数 .如果不存在,则返回数组中最大的数.
测试用例:
输入:[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();
}
}