Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情。
一、题目描述:
414. 第三大的数 - 力扣(LeetCode) (leetcode-cn.com)
给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。
示例 1:
输入:[3, 2, 1]
输出:1
解释:第三大的数是 1 。
示例 2:
输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2 。
示例 3:
输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1 。
提示:
- 1 <= nums.length <= 10^4
- -2^31 <= nums[i] <= 2^31 - 1
进阶: 你能设计一个时间复杂度 O(n) 的解决方案吗?
二、思路分析:
说到寻找第几大的数第一时间就想到了排序,利用内置sort排序,然后unique去重容器将重复元素移至数组尾部,利用erase删除重复元素,最后对数组长度进行判断,如果长度小于3则返回数组最后一个元素,否则返回倒数第三个元素。
三、AC 代码:
class Solution {
public:
int thirdMax(vector<int>& nums) {
sort(nums.begin(),nums.end());
vector<int>::iterator it = unique(nums.begin(),nums.end());
nums.erase(it,nums.end());
if(nums.size() < 3)
{
return nums[nums.size() - 1];
}
return nums[nums.size() - 3];
}
};