Third Maximum Number

Question

Given a non-empty array of integers, return the third maximum number in this array. If it does not exists, return the maximum number. The time complexity must be in O(n).

Note

• 设置3个临时变量，分别存储max1, max2, max3。也可以是一个长度为3的数组。

Extension

• 求第k大的数（k < 5, k值变大，求解效率变低。），都可以使用这个方法来求解。
• 堆排序雏形。求第k大（小），可以建立k长的大顶堆（小顶堆）。
• Java中，PriorityQueue是用数组的形式实现了二叉小顶堆。

Solution

public int thirdMax(int[] nums) {
    Integer max1 = null;
    Integer max2 = null;
    Integer max3 = null;
    for (Integer n : nums) {
        if (n.equals(max1) || n.equals(max2) || n.equals(max3)) {
            continue;
        }
        if (max1 == null || n > max1) {
            max3 = max2;
            max2 = max1;
            max1 = n;
        } else if (max2 == null || n > max2) {
            max3 = max2;
            max2 = n;
        } else if (max3 == null || n > max3) {
            max3 = n;
        }
    }
    return max3 == null ? max1 : max3;
}
// 注：此解法非原创，只供提供学习思路。