414. Third Maximum Number

最新推荐文章于 2024-07-10 23:03:37 发布

kakitgogogo

最新推荐文章于 2024-07-10 23:03:37 发布

阅读量216

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： array

本文链接：https://blog.youkuaiyun.com/kakitgogogo/article/details/54706366

leetcode 专栏收录该内容

230 篇文章

订阅专栏

本文介绍了一种在O(n)时间复杂度内找到数组中第三大数的方法，若不存在则返回最大数。通过使用集合存储当前前三大的数，确保了算法的效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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.

求一个数组中第3大的数，如果不存在第3大的数，则返回最大的数。用set可以实现。遍历数组，将数组元素插入set，并保持set大小为3，如果超过3，就删除set的第一项（最小的）。最后如果set的大小是3的话说明有第3大的数，返回set的第一项；否则的话返回set的最后一项。

代码：

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