414 第三大的数(分情况讨论)

1. 问题描述:

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

示例 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/third-maximum-number

2. 思路分析:

分析题目可以知道我们可以使用三个变量a, b, c来维护最大值,次大值,第三大的值,分情况讨论即可,对于当前的x判断落入到这三个数的哪一个区间来决定更新a,b c中的某些数字。因为相同的数字排名是相同的所以需要使用一个变量count来记录不同数字的个数,当循环结束之后判断count是否小于3如果小于3那么返回最大值即可,否则返回第三大的值。

3. 代码如下:

from typing import List


class Solution:
    def thirdMax(self, nums: List[int]) -> int:
        # 思路是分情况讨论, 初始化为最小值
        a = b = c = -2147483649
        # count用来记录不同数的个数
        count = 0
        for x in nums:
            if x > a:
                # a, b, c都往前移动一位
                c = b
                b = a
                a = x
                count += 1
            elif b < x < a:
                # 更新b, c
                c = b
                b = x
                count += 1
            elif c < x < b:
                # 更新c
                c = x
                count += 1
        if count < 3: return a
        return c


版权声明:本文为qq_39445165原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。