414-第三大的数
题目描述: 给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。
示例1:
输入:[3, 2, 1]
输出:1
解释:第三大的数是 1 。
示例2:
输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2 。
示例3:
输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1 。
提示:
1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1
思路:
- 通过示例1我们可以知道,如果数组长度不够 3 时,直接返回最大值就可以了。
- 通过示例3我们可以知道,‘排序’后的元素中不应该存在重复元素,这时我们应该想到使用set来存储
- set 特征:有序,无重复元素。
- 代码演示
- 这个代码,用增强 for 循环遍历数组,并逐个加入 set ,此处要知道,如果是重复元素,会添加失败。
- for 循环中的 if 判断是为了判断 s 中元素是否大于三个,如果大于三个,我们可以直接删除头元素(最小的那个)此处我们要找第三大的元素,所以保存最大的三个就可以了。
- return 语句:
- 此处前提应该知道,从 for 循环出来,s最大长度为3。
- 使用三元运算符,判断 s 长度是否等于三,如果等于,直接返回s中最前面那个元素,即第三大的元素。如果不满足,直接返回最后那个元素,即最大元素。
class Solution {
public int thirdMax(int[] nums) {
TreeSet<Integer> s = new TreeSet<Integer>();
for (int val : nums) {
s.add(val);
if (s.size() > 3) {
s.remove(s.first());
}
}
return s.size() == 3 ? s.first() : s.last();
}
}