414. 第三大的数——Java

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

示例 1:
输入: [3, 2, 1]
输出: 1
解释: 第三大的数是 1.

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

示例 3:
输入: [2, 2, 3, 1]
输出: 1
解释: 注意，要求返回第三大的数，是指第三大且唯一出现的数。
存在两个值为2的数，它们都排第二。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/third-maximum-number

方法一：一次遍历

public static int thirdMax2(int[] nums) {
    
    
    if (nums.length == 1) {
    
    
        return nums[0];
    }
    if (nums.length == 2) {
    
    
        return Math.max(nums[0], nums[1]);
    }

    int max1 = Integer.MIN_VALUE;
    int max2 = Integer.MIN_VALUE;
    int max3 = Integer.MIN_VALUE;
    int flag = 0;
    boolean f = true;
    for (int i = 0; i < nums.length; i++) {
    
    
        if (nums[i] > max1) {
    
    
            flag++;
            //原先第二大传递给第三大
            max3 = max2;
            //原先最大值传递给第二大
            max2 = max1;
            //更新最大值
            max1 = nums[i];
        } else if (nums[i] > max2 && nums[i] < max1) {
    
    
            flag++;
            max3 = max2;
            max2 = nums[i];
        } else if (nums[i] > max3 && nums[i] < max2) {
    
    
            flag++;
            max3 = nums[i];
        } else if (nums[i] == Integer.MIN_VALUE && f) {
    
    
            flag++;
            f = false;
        }
    }
    return flag >= 3 ? max3 : max1;
}

方法二：若不规定时间复杂度为O(n)可用

public static int thirdMax2(int[] nums) {
    
    
    HashSet<Integer> set = new HashSet<>();
    for (int num : nums) {
    
    
        set.add(num);
    }
    Object[] objects = set.toArray();
    Arrays.sort(objects);//O(nlogn)
    int thirdmax;
    if (set.size() < 3) {
    
    
        thirdmax = (int) objects[objects.length - 1];
    } else {
    
    
        thirdmax = (int) objects[objects.length - 3];
    }
    return thirdmax;
}