剑指offer 61 - 扑克牌中的顺子 - python

271 阅读1分钟

题目描述:

从扑克牌中随机抽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)有几个,因为0小于任何正整数,所以可以先排序,然后看除0之外的数字。依次比较当前数字和下一个数字:

  • 如果相等,必然构不成顺子,直接返回False

  • 如果下一个数字比当前数字大1,则继续往后

  • 如果下一个数字比当前数字大的超过1,则继续两者的差值,表示需要多少个0来填补

    • 如果差值减1大于当前0的个数,则直接返回False
    • 如果差值减1小于当前0的个数,则更新0的个数继续往后
class Solution:
    def isStraight(self, nums: List[int]) -> bool:
        if nums == []: return False

        zero = nums.count(0)
        nums.sort()
        count = 0
        for i in range(zero, len(nums) - 1):
            if nums[i] == nums[i + 1]:
                return False
            elif nums[i] + 1 == nums[i + 1]:
                continue
            else:
                zero = zero - (nums[i + 1] - nums[i] - 1)
                if zero < 0:
                    return False

        return True

在这里插入图片描述