第三大的数

给定一个非空数组，返回此数组中第三大的数。如果不存在，则返回数组中最大的数。要求算法时间复杂度必须是O(n)。

• 示例 1:
  输入: [3, 2, 1]
  输出: 1
  解释: 第三大的数是 1.
• 示例 2:
  输入: [1, 2]
  输出: 2
  解释: 第三大的数不存在, 所以返回最大的数 2 .

我的方案

思路

原本我使用了Arrays.sort()方法，但是这种堆排序的时间复杂度是O(nlogn)，不符合题意。但是这道题最好是先进行排序，所以我想到了TreeSet这种红黑树。

算法

遍历整个数组时，不断加入红黑树进行排序，并将红黑树保持3个大小。遍历完后，如果红黑树等于三个表示数组有第三大的数，输出。反之输出最大数。

代码

class Solution {
    public int thirdMax(int[] nums) {
        TreeSet<Integer> set = new TreeSet<Integer>();
        for(int i = 0;i<nums.length;i++){
            set.add(nums[i]);
            //这里做一个红黑树的活动窗口，每当窗口扩张到4个时，将最小的清除
            if(set.size()>3){
                set.pollFirst();
            }
            
        }
        if(set.size()==3){
            return set.first();
        }
        return set.last();
    }
}

精选方案

思路

1. 用三个变量来存放第一大，第二大，第三大的元素的变量，分别为one, two, three；
2. 遍历数组，若该元素比one大则往后顺移一个元素，比two大则往后顺移一个元素，比three大则赋值个three；
3. 最后得到第三大的元素，若没有则返回第一大的元素。

作者：happy_yuxuan

代码

class Solution {
    private long MIN = Long.MIN_VALUE;    // MIN代表没有在值
    
    public int thirdMax(int[] nums) {
        if (nums == null || nums.length == 0) throw new RuntimeException("nums is null or length of 0");
        int n = nums.length;
        
        int one = nums[0];
        long two = MIN;
        long three = MIN;
        
        for (int i = 1; i <  n; i++) {
            int now = nums[i];
            // 如果当前值是已判断过，跳过
            if (now == one || now == two || now == three) continue;    
            if (now > one) {
                three = two;
                two = one;
                one = now;
            } else if (now > two) {
                three = two;
                two = now;
            } else if (now > three) {
                three = now;
            }
        }

        if (three == MIN) return one;  // 没有第三大的元素，就返回最大值
        return (int)three;
    }
}

作者：happy_yuxuan