[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.

分析：

返回数组中第三大的数，不存在就返回最大的数。先对数组中的数排序，找出最大的三个数放入新建的容器res中，注意res中的数不可重复，因此需事先判断。res长度为3或者原数组全部遍历一遍退出循环，若res长度为3，表示存在第三大的数，返回res[2]，否则返回res[0]。

class Solution {
public:
    int thirdMax(vector<int>& nums) {
        sort(nums.begin() , nums.end());
        vector<int> res;
        int len = res.size();
        for(int i=nums.size()-1; i>=0 && len<3; i--)
        {
            if(res.empty() || (nums[i]!=res.back()))
            {
               res.push_back(nums[i]);
                len++;
            }              
        }
        if(len == 3)
            return res[2];
        else
            return res[0];              
    }
};