LeetCode——第三大的数

Q：给定一个非空数组，返回此数组中第三大的数。如果不存在，则返回数组中最大的数。要求算法时间复杂度必须是O(n)。
示例 1:
输入: [3, 2, 1]
输出: 1
解释: 第三大的数是 1.

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

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

A：
1.用set最方便，直接Treeset排序好。

    public int thirdMax(int[] nums) {
        if (nums.length == 0)
            return 0;
        Set<Integer> set = new TreeSet<>(Collections.reverseOrder());
        for (int i : nums) {
            set.add(i);
        }
        int count = 3;
        int res = nums[0];
        Iterator<Integer> it = set.iterator();
        if (set.size() < 3)
            return it.next();
        while (count-- != 0) {
            res = it.next();
        }
        return res;
    }

2.设置三个max

    public int thirdMax(int[] nums) {
        if (nums.length == 0)
            return 0;
        long max1 = Long.MIN_VALUE, max2 = Long.MIN_VALUE, max3 = Long.MIN_VALUE;//不要设置为Integer.MIN_VALUE，如果里面出现了就麻烦
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] > max1) {
                max3 = max2;
                max2 = max1;
                max1 = nums[i];
            } else if (nums[i] > max2 && nums[i] != max1) {
                max3 = max2;
                max2 = nums[i];
            } else if (nums[i] > max3 && nums[i] != max2 && nums[i] != max1) {//要记得，num[i]不能等于任何已经出现过的max值
                max3 = nums[i];
            }
        }
        if (max3 == Long.MIN_VALUE) {
            return (int)max1;
        }
        return (int)max3;
    }