題目描述

給定一個非空數組，返回此數組中第三大的數。如果不存在，則返回數組中最大的數。要求算法時間復雜度必須是O(n)。

示例

示例 1:

輸入: [3, 2, 1] 輸出: 1 解釋: 第三大的數是 1.

示例 2:

輸入: [1, 2] 輸出: 2 解釋: 第三大的數不存在, 所以返回最大的數 2 .

示例 3:

輸入: [2, 2, 3, 1] 輸出: 1 解釋: 注意，要求返回第三大的數，是指第三大且唯一出現的數。 存在兩個值為2的數，它們都排第二。

分析

保持一個大小為3的集合。
遍歷數組，然后將元素插入，如果集合大小小于等于3，則繼續，如果大于3，則取消set中第一個元素。接著遍歷。

分析：
給一個最大為3的插入，刪除一個元素的時間復雜度為O(nlogn)，在這里即為O(3)。所以時間負責度為O(n)

實現

class Solution { public:int thirdMax(vector<int>& nums) {set<int> s;for (int num : nums) {s.insert(num);if (s.size() > 3) {s.erase(s.begin());}}return s.size() == 3 ? *s.begin() : *s.rbegin();} };

總結

以上是生活随笔為你收集整理的【LeetCode】414.第三大的数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。