给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/third-maximum-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例 1:
输入:[3, 2, 1]
输出:1
解释:第三大的数是 1 。
示例 2:
输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2 。
示例 3:
输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1 。
题解: 这道题很简单
注意,这道题有点坑,我们这里需要去重,排序最好使用set去重+排序
class Solution
{
public:int thirdMax(vector &nums){if (nums.empty() == true)return -1;// 找第三个大的数字 -- 建立小堆priority_queue, greater> q; // 建立小对unordered_set set;for (auto e : nums){set.insert(e);}unordered_set::iterator it = set.begin();int i = 0;while (it != set.end()){if (i < 3){// 入堆q.push(*it);i++;}else{// 判断 如果 当前数据大于堆顶int ret = q.top();if (*it > ret){q.pop();q.push(*it);}}++it;}// for (size_t i = 0; i < nums.size(); i++)// {// if (i < 3)// {// // 入堆// q.push(nums[i]);// }// else// {// // 判断 如果 当前数据大于堆顶// int ret = q.top();// if (nums[i] > ret)// {// q.pop();// q.push(nums[i]);// }// }// }if (q.size() != 3){while (q.size() != 1){q.pop();}}return q.top();}
};