第三大的数(C++)

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

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

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