leetcode-数组-简单-第三大的数

题目

给定一个非空数组，返回此数组中第三大的数。如果不存在，则返回数组中最大的数。要求算法时间复杂度必须是O(n)。

来源：

https://leetcode-cn.com/problems/third-maximum-number/

示例：

示例 1:
输入: [3, 2, 1]
输出: 1
解释: 第三大的数是 1.
示例 2:
输入: [1, 2]
输出: 2
解释: 第三大的数不存在, 所以返回最大的数 2 .
示例 3:
输入: [2, 2, 3, 1]
输出: 1
解释: 注意，要求返回第三大的数，是指第三大且唯一出现的数。
存在两个值为2的数，它们都排第二。

解答

方法一：

分析如下：

• 以前求最大值的时候，是用一个数来记录最大值，然后挨个对比更新。
• 那么第三大的数，就用三个数来记录，分别是第一，二，三大的数。
• 如果第一大的数更新了，那么原来的二三也要更新
• 如果第二大更新了，那么原来的三要更新。
• 返回的时候处理一下相等的情况

class Solution {
public:
    int thirdMax(vector<int>& nums) {
        if (nums.size() == 1) {
            return nums[0];
        }
        if (nums.size() == 2) {
            return std::max(nums[0], nums[1]);
        }
        //以下是>= 3的情形
        std::array<int, 3> max_arr = {nums[0], nums[1], nums[2]};
        std::sort(max_arr.begin(), max_arr.end());
        for (int i = 3; i < nums.size(); ++i) {
            //每一个数字放到数组里进行对比,比任何一个数大都要进行替换
            //先比较第一大
            if (nums[i] >= max_arr[2]) {
                if (nums[i] == max_arr[2]) {
                    continue;
                }
                //当最大值更新的时候，第二大和第三大都要更新一下
                max_arr[0] = max_arr[1];
                max_arr[1] = max_arr[2];
                max_arr[2] = nums[i];               
            } else if (nums[i] >= max_arr[1]) {
                //第二大更新的时候，第三大要更新一下
                if (nums[i] == max_arr[1]) {
                    continue;
                }
                max_arr[0] = max_arr[1];
                max_arr[1] = nums[i];
            } else if (nums[i] >= max_arr[0]) {
                max_arr[0] = nums[i];
            } else {
                if (max_arr[0] != max_arr[1]) {
                    continue;
                }
                //因为如果 第三和第二相等，不处理比三小的，那么会遗漏真正的第三
                max_arr[0] = nums[i];
            }
        }
        /// 三种情形   111, 112, 122
        if (max_arr[0] == max_arr[1] || max_arr[1] == max_arr[2]) {
            return max_arr[2];
        } 
        return max_arr[0];
    }
};