好记性不如烂笔头的专栏

每天进步一点点

Leetcode-414. Third Maximum Number

题目

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.

解答

这一题看着简单,其实不是那么容易AC的。主要是需要对一些特殊的输入进行特殊处理。
注意:[2,1,1]这种情况是不存在第三大的,返回2。

用三个变量first,second,third来表示第一大,第二大,第三大。
用count来表示这三个更新次数。如果更新次数大于等于3,直接返回third。
如果小于3,那就是返回最大值。
但是有一个特殊的例子:
[2,1,-2147483648],这个在判断第三个时是不更新的,所以额外对其进行计数,用bottom来计数,在最后的判断中就用count+bottom来判断。
AC代码:9ms

class Solution {
public:
    int thirdMax(vector<int>& nums) {
        #define INT_MIN (-2147483647-1)
        int max_3[3];
        if(nums.size()==1) return nums[0];
        else if(nums.size()==2) return nums[1]>nums[0]?nums[1]:nums[0];
        else
        {
            int first=INT_MIN,second=INT_MIN,third=INT_MIN;
            bool flag1=0,flag2=0,flag3=0;//是否顶替掉初值
            int count=0;
            int bottom=0;

            //缺少对三个一样的进行判断
            for(int i=0;i<nums.size();i++)
            {
                if(nums[i]==-2147483648) bottom=1;//有一个不需要更新
                if(nums[i]==first||nums[i]==second||nums[i]==third)//如果有相同的数,至少有一个是更新的
                    continue;
                if(nums[i]>first)
                {
                    third=second;
                    second=first;
                    first=nums[i];
                    count++;
                }
                else if(nums[i]>second)
                {
                    third=second;
                    second=nums[i];
                    count++;
                }
                else if(nums[i]>third)
                {
                    third=nums[i];
                    count++;
                }
            }
            if(count+bottom>=3)// 
                return third;//三个都更新了
            else 
                return first;//两个更新,一个更新或者都没更新,没达到要求,返回第一个值

        }

    }
};
阅读更多
版权声明:转载标明出处,点一个顶即可:) https://blog.csdn.net/starstar1992/article/details/53983947
个人分类: 算法
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

Leetcode-414. Third Maximum Number

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭