剑指 Offer 61. 扑克牌中的顺子

题目描述

https://leetcode-cn.com/problems/bu-ke-pai-zhong-de-shun-zi-lcof
从扑克牌中随机抽5张牌，判断是不是一个顺子，即这5张牌是不是连续的。2～10为数字本身，A为1，J为11，Q为12，K为13，而大、小王为 0 ，可以看成任意数字。A 不能视为 14。

示例 1:

输入: [1,2,3,4,5]
输出: True

示例 2:

输入: [0,0,1,2,5]
输出: True

限制：

数组长度为 5 
数组的数取值为 [0, 13] .

解题思路

使用贪心算法解决。
对数组进行排序。
首先先看以后没有除了0之外的重复牌，有的话直接False。
然后看有几张0，记住count_0，并去除所有0。
接下来，从头遍历数组，缺如果中间有缺额就补0，只要0够用，就是顺子。

解题代码

class Solution:
    def isStraight(self, nums: list[int]) -> bool:
        nums.sort()
        # 如果有除了大小王之外的重复牌,直接False
        for i in range(len(nums) - 1):
            if nums[i] != 0 and nums[i] == nums[i + 1]:
                return False
        # 获得0的数量
        count_0 = nums.count(0)
        # 去除所有的0
        for i in range(count_0):
            nums.remove(0)
        # 判断大小王能不能填满空位
        for i in range(len(nums) - 1):
            count_0 = count_0 - (nums[i+1] - nums[i] - 1)
        if count_0 >= 0:
            return True
        else:
            return False

执行结果

56ms 14.6MB