解题思路:
第一种方法就是利用排序去解决,首先按照从小到大的顺序排列,接着从最后一个数出发,用一个变量标记不同的数的个数,一旦找到第三个就返回,如果实在找不到就返回最大的,代码如下:
class Solution {
public:
int thirdMax(vector<int>& nums) {
sort(nums.begin(), nums.end());
int count = 1;
for(int i = nums.size() - 1; i > 0; i --) {
if(nums[i] != nums[i - 1] && ++count == 3) {
return nums[i - 1];
}
}
return nums[nums.size() - 1];
}
};
第二种方法就是O(n)的方法,定义三个变量存储最大、第二大、第三大的数,代码如下:
class Solution {
public:
int thirdMax(vector<int>& nums) {
long a = LONG_MIN, b = LONG_MIN, c = LONG_MIN;
for(long num : nums) {
if(num > a) {
c = b;
b = a;
a = num;
} else if(a > num && num > b) {
c = b;
b = num;
} else if(b > num && num > c) {
c = num;
}
}
return c > LONG_MIN ? c : a;
}
};