第三大的数(C++)

第三大的数

难度简单385收藏分享切换为英文接收动态反馈

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

示例 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

我的代码：

class Solution {
public:
// set排序加去重 然后找到倒数第三个就可以了
    int thirdMax(vector<int>& nums) {
        int ma = 0;
        set<int> st;
        for (int i = 0; i < nums.size(); ++ i)
        {
            st.insert(nums[i]);
            ma = max(ma, nums[i]);
        } 
        if (st.size() >= 3) 
        {
            int cnt = 0;
            for(auto s : st)
            {
                if (cnt == st.size() - 3) return s;
                cnt ++;
            }
        }   
        return ma;
    }
};

对应我的掘金文章：https://juejin.cn/post/7147265935374549000