爱丽丝有一手(hand
)由整数数组给定的牌。
现在她想把牌重新排列成组,使得每个组的大小都是 W
,且由 W
张连续的牌组成。
如果她可以完成分组就返回 true
,否则返回 false
。
注意:此题目与 1296 重复:https://leetcode-cn.com/problems/divide-array-in-sets-of-k-consecutive-numbers/
示例 1:
输入:hand = [1,2,3,6,2,3,4,7,8], W = 3
输出:true
解释:爱丽丝的手牌可以被重新排列为 [1,2,3],[2,3,4],[6,7,8]
。
示例 2:
输入:hand = [1,2,3,4,5], W = 4 输出:false 解释:爱丽丝的手牌无法被重新排列成几个大小为 4 的组。
提示:
1 <= hand.length <= 10000
0 <= hand[i] <= 10^9
1 <= W <= hand.length
class Solution {
public boolean isNStraightHand(int[] hand, int groupSize) {
if (hand.length % groupSize != 0) {
return false;
}
TreeMap<Integer, Integer> mp = new TreeMap<>();
for (int item : hand) {
mp.put(item, mp.getOrDefault(item, 0) + 1);
}
while (mp.size() > 0) {
int start = mp.firstKey();
for (int i = start; i < start + groupSize; i++) {
if (!mp.containsKey(i)) {
return false;
}
int time = mp.get(i);
if (time == 1) {
mp.remove(i);
} else {
mp.replace(i, time - 1);
}
}
}
return true;
}
}