给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。
示例 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
给数组排序,从右往前遍历,初始化最小值为数组中最右边的值,如果左边一个数小于当前最小值,则计数器flag++,判断flag的值如果大于三说明有第三大数,如果循环结束flag小于三,说明没有,返回数组中最大值
class Solution {
public int thirdMax(int[] nums) {
int flag = 1;
Arrays.sort(nums);//排序
int min = nums[nums.length - 1];
for(int i = nums.length - 1;i > 0;i--) {
if(nums[i] > nums[i - 1]) {
min = nums[i - 1];//从右往左遍历
flag++;//计数器++
}
if(flag >= 3) {
return min;//找到第三大的数
}
}
if(flag < 3) {
return nums[nums.length - 1];//返回最大值
}
return 1;
}
}
题解里这个树还没用过(cv)
class Solution {
public int thirdMax(int[] nums) {
TreeSet<Integer> s = new TreeSet<Integer>();
for (int num : nums) {
s.add(num);
if (s.size() > 3) {
s.remove(s.first());
}
}
return s.size() == 3 ? s.first() : s.last();
}
}