关闭

Leetcode-414. Third Maximum Number

145人阅读 评论(0) 收藏 举报
分类:

题目

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;//两个更新,一个更新或者都没更新,没达到要求,返回第一个值

        }

    }
};
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:54240次
    • 积分:2012
    • 等级:
    • 排名:千里之外
    • 原创:149篇
    • 转载:21篇
    • 译文:0篇
    • 评论:9条
    最新评论