Leetcode. 414. 第三大的数

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
 

进阶：你能设计一个时间复杂度 O(n) 的解决方案吗？

方法一：排序

• 我们将数组排序后，对其进行遍历取第三大即可，注意要去重。

class Solution {
    public int thirdMax(int[] nums) {
        Arrays.sort(nums);
        int count = 2;
        boolean flag = false;

        for(int i=nums.length-2;i>=0;i--){
            if(nums[i]==nums[i+1]){
            }
            else{
                count--;
                if(count==0){
                    count = nums[i];
                    flag = true; 
                    break;
                }
            }
        }
        if(!flag){
            return nums[nums.length-1];
        }

        return count;
    }
}

方法二：使用有序集合

• 我们使用有序且存储不重复数据的TreeSet集合即可实现，即将nums的元素放到TreeSet集合中即可自己进行筛选，且元素放置时会按照大小顺序进行排序。

class Solution {
    public int thirdMax(int[] nums) {
        TreeSet<Integer> set = new TreeSet<>();
        for(int i=0;i<nums.length;i++){
            set.add(nums[i]);
            if(set.size()>3){
                set.remove(set.first());
            }
        }

        if(set.size() == 3){
            return set.first();
        }
        return set.last();
    }
}

方法三：模拟有序集合

• 同方法二一样的思想，不过不使用java的集合，而是自己手动实现，即我们可以通过创建一个长度为3的数组，也可以创建三个变量，三个元素的意义分别为存储最大元素，第二大元素，第三大元素，接着遍历nums，一旦发现符合要求的则对三个元素进行更新即可。最后根据三个元素的填满情况来判断返回最大还是第三大元素。

class Solution {
    public int thirdMax(int[] nums) {
        long pre = Long.MIN_VALUE;
        long mid = Long.MIN_VALUE;
        long next = Long.MIN_VALUE;
        for (int i=0;i<nums.length;i++) {
            if (nums[i] > pre) {
                next = mid;
                mid = pre;
                pre = nums[i] ;
            } else if (pre > nums[i]  && nums[i]  > mid) {
                next = mid;
                mid = nums[i] ;
            } else if (mid > nums[i]  && nums[i]  > next) {
                next = nums[i] ;
            }
        }

        if(next == Long.MIN_VALUE ){
            return (int)pre;
        }

        return (int)next;
    }
}