题目描述:
给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。
示例 1:
输入:[3, 2, 1]
输出:1
解释:第三大的数是 1 。
示例 2:输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2 。
示例 3:输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1 。
方法一:排序(C++代码)
class Solution {
public:
int thirdMax(vector<int>& nums) { //方法一:排序
sort(nums.begin(), nums.end(), greater<>()); //从大到小
for(int i = 1,count = 1; i < nums.size(); i++){
if(nums[i] != nums[i - 1] && ++count == 3){ //注意:先判断&&之前的,若不符合条件,则直接跳出if,不会再判断&&后面的表达式
return nums[i]; //此时nums[i]就是第三大的数
}
}
return nums[0];
}
};
时间复杂度:O(nlogn),其中 n 是数组 nums 的长度。排序需要 O(nlogn) 的时间。
空间复杂度:O(logn),排序需要的栈空间为 O(logn)。