414.第三大的数

题目描述
简单解法思路O(nlogn)
代码
O(n)思路
代码

toc

题目描述

给你一个非空数组，返回此数组中 第三大的数 。如果不存在，则返回数组中最大的数。

示例 1：

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

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

输入：[2, 2, 3, 1]
输出：1
解释：注意，要求返回第三大的数，是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数，它们都排第二。在所有不同数字中排第三大的数为 1 。

提示：

1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1

进阶：你能设计一个时间复杂度 O(n) 的解决方案吗？

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

简单解法思路O(nlogn)

通过stl库自带的有序容器将输入排序，并将容器大小控制在4个以下，最后容器大小为3，则输出容器内最小的数，否则输出最大的数

代码

class Solution {
public:
    int thirdMax(vector<int>& nums) {
        if(nums.empty())
            return 0;

        std::set<int> sortSet;
        for(const auto& elem : nums){
            sortSet.emplace(elem);    //O(logn)
            if(sortSet.size() > 3)
                sortSet.erase(sortSet.begin());
        }
        return 3 == sortSet.size() ? *sortSet.begin() : *sortSet.rbegin();
    }
};

O(n)思路

直接维护三个变量存储可能存在的最大三个数就可以实现O(n)
注意：

• 题目要求里：不存在第三大数时，返回的是最大数
• 存储最大三个数的变量类型需为有符号long long，这样才能判断出第三大的数是否存在

代码

class Solution {
public:
    int thirdMax(vector<int>& nums) {
        if(nums.empty())
            return 0;

        long long iMin = std::numeric_limits<long long>::lowest();
        long long iFirst = iMin, iSecond = iMin, iThird = iMin;
        for(const auto& elem : nums){
            if(elem == iFirst || elem == iSecond || elem == iThird)
                continue;

            if(elem > iFirst){
                iThird = iSecond;
                iSecond = iFirst;
                iFirst =elem;
                continue;
            }else if(elem > iSecond){
                iThird = iSecond;
                iSecond = elem;
                continue;
            }else if(elem > iThird){
                iThird = elem;
                continue;
            }
        }
        return iMin == iThird ? iFirst : iThird;
    }
};

来自为知笔记(Wiz)