【面试题61 扑克牌中的顺子】
面试题61:扑克牌中的顺子:从扑克牌中随机抽5张牌,判断是不是一个顺子,即这5张牌是不是连续的。2~10为数字本身,A为1,J为11,Q为12,K为13,而大、小王为 0 ,可以看成任意数字。A 不能视为 14。
思路:首先对数组排序,统计大小王的个数(讲道理最多两个0,但是题目没说只有一副牌,所有还是有可能有多个0的)。从第二个不为 0 的数开始检查顺子,计算和前一个不为 0 的数之差,并用大小王来补(因为大小王可以是任何数字),如果能补全所有的不连续点,就返回 True,否则 False。
class Solution:
def isStraight(self, nums: List[int]) -> bool:
nums.sort()
n = len(nums)
count = 0
# 统计大小王个数
for k in nums:
if k == 0: count += 1
# 检查顺子
i = count+1
while i < n:
dif = nums[i] - nums[i-1] - 1 # 不连续时的断点个数
if dif < 0: return False # 有重复数字,肯定不连续
if dif >= 1: count -= dif # 不连续,用大小王(0)来补
if count < 0: return False # 大小王不够用,则一定不连续
i += 1
return True