LeetCode 414. 第三大的数

目录结构

1.题目

2.题解

2.1使用TreeSet

2.2使用三个变量

---

1.题目

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

示例：

入: [3, 2, 1]

输出: 1

解释: 第三大的数是 1.


输入: [1, 2]

输出: 2

解释: 第三大的数不存在, 所以返回最大的数 2 .


输入: [2, 2, 3, 1]

输出: 1

解释: 注意，要求返回第三大的数，是指第三大且唯一出现的数。
存在两个值为2的数，它们都排第二。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/third-maximum-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

2.1使用TreeSet

TreeSet为有序（自然排序为升序）的set集合。

public class Solution414 {

    @Test
    public void test414() {
        int[] nums = {1, 2, 2, 3, 4};
        System.out.println(thirdMax(nums));
    }

    public int thirdMax(int[] nums) {
        TreeSet<Integer> set = new TreeSet<>();
        for (int n : nums) {
            set.add(n);
            if (set.size() > 3) {
                set.remove(set.first());
            }
        }
        return set.size() == 3 ? set.first() : set.last();
    }
}

• 时间复杂度：
• 空间复杂度：

2.2使用三个变量

这里变量使用Long的原因是为了初始化取更小的值，测试数据含Integer.MIN_VALUE。

public class Solution414 {

    @Test
    public void test414() {
        int[] nums = {1, 2, 2, 3, 4};
        System.out.println(thirdMax(nums));
    }

    public int thirdMax(int[] nums) {
        long[] t = {nums[0], Long.MIN_VALUE, Long.MIN_VALUE};
        for (int num : nums) {
            if (num == t[0] || num == t[1] || num == t[2]) {
                continue;
            }
            if (num > t[0]) {
                t[2] = t[1];
                t[1] = t[0];
                t[0] = num;
            } else if (num > t[1]) {
                t[2] = t[1];
                t[1] = num;
            } else if (num > t[2]) {
                t[2] = num;
            }
        }
        return (int) (t[2] == Long.MIN_VALUE ? t[0] : t[2]);
    }
}

• 时间复杂度：
• 空间复杂度：