题目
Alice 手中有一把牌,她想要重新排列这些牌,分成若干组,使每一组的牌数都是 groupSize ,并且由 groupSize 张连续的牌组成。
给你一个整数数组 hand 其中 hand[i] 是写在第 i 张牌,和一个整数 groupSize 。如果她可能重新排列这些牌,返回 true ;否则,返回 false 。
示例
-
示例一
输入:hand = [1,2,3,6,2,3,4,7,8], groupSize = 3
输出:true
解释:Alice 手中的牌可以被重新排列为 [1,2,3],[2,3,4],[6,7,8]。 -
示例二
输入:hand = [1,2,3,4,5], groupSize = 4
输出:false
解释:Alice 手中的牌无法被重新排列成几个大小为 4 的组。
方法
模拟
class Solution {
public boolean isNStraightHand(int[] hand, int m) {
Map<Integer, Integer> map = new HashMap<>();
PriorityQueue<Integer> q = new PriorityQueue<>((a,b)->a-b);
for (int i : hand) {
map.put(i, map.getOrDefault(i, 0) + 1);
q.add(i);
}
while (!q.isEmpty()) {
int t = q.poll();
if (map.get(t) == 0) continue;
for (int i = 0; i < m; i++) {
int cnt = map.getOrDefault(t + i, 0);
if (cnt == 0) return false;
map.put(t + i, cnt - 1);
}
}
return true;
}
}
使用HashMap记录所有数字的出现次数,再使用小根堆存放所有数字。
每次从小根堆中取出堆顶数字,作为一个顺子的第一个数字。
如果此时map里没有这个数字,说明之前被使用过了,继续下一循环。
如果可以作为顺子的起点,则按照顺子长度m,判断map中是否有后续数字,若有则使用,然后再map中将数字个数-1,如果map中没有,则无法被分成若干个长度为 m的顺子。
当小根堆为空时,说明所有数字已经顺利分为若干个顺子了,返回true。