题目
从扑克牌中随机抽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
思路
大小王可以看成任意数字,我们就先把它们都看成0。
判断五个数字是不是连续的,最直观的办法是把数组排序。由于0可以实任意数字,因此我们可以用0去补满数组中的空缺。如果排序之后的数组不是连续的,只要我们有足够的0去补满空缺,这个数字实际上还是连续的。
要做的三件事:
1、统计数组中0的个数
2、统计排序之后的数组中相邻数字间的空缺数
3、如果空缺总数小于等于0的个数,那这个数组就是连续的,否则不连续。
注意:如果非0数字重复出现,则数组不是连续的。
代码
class Solution:
def isStraight(self, nums: List[int]) -> bool:
nums.sort()
zero_count,gap_count=0,0
for num in nums:
if num==0: zero_count+=1
else: break
small = zero_count
big = small+1
while small<big and big<len(nums):
if nums[small]==nums[big]: return False
gap_count = nums[big]-nums[small]-1
small = big
big+=1
if gap_count>zero_count:return False
return True
复杂度
时间复杂度O(nlogn) : 其中 N 为 nums 长度,数组排序使用 O(nlogn) 时间。
空间复杂度 O(1)