LeetCode 846. Hand of Straights
考点 | 难度 |
---|---|
Greedy | Easy |
题目
Alice has some number of cards and she wants to rearrange the cards into groups so that each group is of size groupSize, and consists of groupSize consecutive cards.
Given an integer array hand where hand[i] is the value written on the ith card and an integer groupSize, return true if she can rearrange the cards, or false otherwise.
思路
用map记录每个card的个数。对于每张卡,数groupsize位,如果有任何一位的个数是0返回false,否则个数-1.
答案
class Solution(object):
def isNStraightHand(self, hand, groupSize):
count_map = {}
for card in hand:
count_map[card] = count_map.get(card, 0) + 1
hand.sort()
for i in range(len(hand)):
if count_map[hand[i]] == 0:
continue
for j in range(groupSize):
curr_card = hand[i] + j
if count_map.get(curr_card, 0) == 0:
return False
count_map[curr_card] -= 1
return True