LeetCode 414 第三大的数[模拟 排序] HERODING的LeetCode之路

Posted 2021-10-25 HERODING23

解题思路：
第一种方法就是利用排序去解决，首先按照从小到大的顺序排列，接着从最后一个数出发，用一个变量标记不同的数的个数，一旦找到第三个就返回，如果实在找不到就返回最大的，代码如下：

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;
    }
};