LeetCode 414. 第三大的数

123 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述:

414. 第三大的数 - 力扣(LeetCode) (leetcode-cn.com)

给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。

示例 1:

输入:[3, 2, 1]
输出:1
解释:第三大的数是 1

示例 2:

输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2

示例 3:

输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1

提示:

  • 1 <= nums.length <= 10^4
  • -2^31 <= nums[i] <= 2^31 - 1

进阶: 你能设计一个时间复杂度 O(n) 的解决方案吗?

二、思路分析:

说到寻找第几大的数第一时间就想到了排序,利用内置sort排序,然后unique去重容器将重复元素移至数组尾部,利用erase删除重复元素,最后对数组长度进行判断,如果长度小于3则返回数组最后一个元素,否则返回倒数第三个元素。

三、AC 代码:

class Solution {
public:
  int thirdMax(vector<int>& nums) {
    sort(nums.begin(),nums.end());
    vector<int>::iterator it = unique(nums.begin(),nums.end());
    nums.erase(it,nums.end());
    if(nums.size() < 3)
    {
      return nums[nums.size() - 1];
    }
    return nums[nums.size() - 3];
  }
};

范文参考:

第三大的数 - 第三大的数 - 力扣(LeetCode) (leetcode-cn.com)