Given a non-empty array of integers, return the third maximum number in this array. If it does not exist, return the maximum number. The time complexity must be in O(n).
Example 1:
Input: [3, 2, 1] Output: 1 Explanation: The third maximum is 1.
Example 2:
Input: [1, 2] Output: 2 Explanation: The third maximum does not exist, so the maximum (2) is returned instead.
Example 3:
Input: [2, 2, 3, 1] Output: 1 Explanation: Note that the third maximum here means the third maximum distinct number. Both numbers with value 2 are both considered as second maximum.
先排序,然后设置计数器找到第三个大的元素。注意如果找不到,那么就返回最大的那个元素。
class Solution {
public:
int thirdMax(vector<int>& nums) {
sort( nums.begin(), nums.end() );
int len = nums.size();
int maxNumber = nums[len - 1];
int cnt = 1;
int curMax = maxNumber;
int thirdMax = maxNumber;
for( int i = len - 1; i >= 0 ; i-- ) {
if( nums[i] != curMax ) {
curMax = nums[i];
cnt++;
}
if( cnt == 3 ) {
thirdMax = curMax;
break;
}
}
return thirdMax;
}
};