题目描述
给你一个非空数组，返回此数组中 第三大的数 。如果不存在，则返回数组中最大的数。
示例 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;
}
};