一、运行结果
二、题目
给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。
示例 1:
输入:[3, 2, 1]
输出:1
解释:第三大的数是 1 。
示例 2:
输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2 。
示例 3:
输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/third-maximum-number
三、思路
在示例三中有说明,要求返回第三大的数,是指在所有不同数字中排第三大的数。用三个变量first,second,third分别保存当前第一大,第二大,第三大的数,first初始化为数组第一个值,second和third都初始化为INT_MIN,并设置一个变量flag标志数组中是否出现过INT_MIN,根据flag的值可以判断second和third是否更改过。然后遍历数组,根据当前数组元素的大小,修改first,second,third的值,具体修改过程都在代码中有注释。最后返回结果时,如果third从始至终都没有修改过(或重新赋值),则表示第三大元素不存在,返回最大元素,否则返回third。
四、代码
class Solution {
public:
int thirdMax(vector<int>& nums) {
int len = nums.size();
if(len == 1) return nums[0];
if(len == 2) return max(nums[0], nums[1]);
int first = nums[0], second = INT_MIN, third = INT_MIN;
bool flag = false; //标记最小整数是否出现过
for(int i=0; i<len; i++){
if(nums[i] == INT_MIN) flag = true;
if(nums[i] > first){
if(second != INT_MIN){ //第二大的值已经变过,三个都要变
third = second;
second = first;
first = nums[i];
}else{ //第二大的值还没变过,第三大也不能变
second = first;
first = nums[i];
}
}else if(nums[i] < first && nums[i] > second){
if(second != INT_MIN){ //第二大的值已经变过,两个都要变
third = second;
second = nums[i];
}else{
second = nums[i];
}
}else if(nums[i] < second && nums[i] > third){
third = nums[i];
}
}
// 如果第二大和第三大相等或没有出现过INT_MIN而第三大依然为该值(说明第三大没有变过)
if(second == third || (flag==false && third == INT_MIN)) return first;
return third;
}
};