题目:
给定一个非空数组,返回此数组中第三大的数。如果不存在,则返回数组中最大的数。要求算法时间复杂度必须是O(n)。
示例 1:
输入: [1, 2]
输出: 2
示例 2:
输入: [2, 2, 3, 1]
输出: 1
class Solution {
public:
int thirdMax(vector<int>& nums) {
int i, size = nums.size();
int * second_Max = nullptr, *third_Max = nullptr, *first_Max = &nums[0];
for (i = 1; i < size; i++){
if (nums[i] > * first_Max){
third_Max = second_Max;
second_Max = first_Max;
first_Max = &nums[i];
}
else if (nums[i] != *first_Max
&& (second_Max == nullptr || nums[i] > * second_Max)){
third_Max = second_Max;
second_Max = &nums[i];
}
else if (nums[i] != *first_Max && nums[i] != *second_Max
&& (third_Max == nullptr || nums[i] > * third_Max)){
third_Max = &nums[i];
}
}
if (second_Max == nullptr) return *first_Max;
return (third_Max == nullptr) ? max(*second_Max, *first_Max): *third_Max;
}
};
题解:
定义三个变量存放first_Max,second_Max和third_Max三个数,最开始将第一个元素赋值给first_Max,对数组从下标1开始遍历。
①若数字大于first_Max,则更新first_Max,second_Max和third_Max。
②若数字不等于first_Max并且,second_Max不存在或者大于second_Max,更新second_Max和third_Max。
③若数字不等于first_Max并且不等于second_Max并且,third_Max不存在或者大于third_Max,更新third_Max。
最后判断返回值:若second_Max不存在,则返回first_Max;若third_Max不存在,则返回first_Max与second_Max中最大的;否则返回third_Max。