感覺應該還可以剪一些枝
懶剪了
def isNStraightHand(self, hand: List[int], groupSize: int) -> bool:
cnt = defaultdict(int)
for i in hand:
cnt[i] += 1
sorted_keys = sorted(cnt.keys())
for k in sorted_keys:
cnt_cur = cnt[k]
if cnt_cur>0:
for i in range(groupSize):
if (k+i) in cnt.keys():
cnt[k+i] = cnt[k+i]-cnt_cur
else:
return False
elif cnt_cur<0:
return False
return True