爱丽丝有一手(hand)由整数数组给定的牌。
现在她想把牌重新排列成组,使得每个组的大小都是 W,且由 W 张连续的牌组成。
如果她可以完成分组就返回 true,否则返回 false。
示例 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
思路:将每种牌对应的数量存入map中,然后对所有牌按照大小升序排序,从第一个开始遍历,每次看往后连续W个数是否存在,没有或者已经用完直接返回false,如果所有牌刚好凑完则返回true。复杂度O(nlogn)。
class Solution {
public boolean isNStraightHand(int[] hand, int W) {
int n=hand.length;
if(n%W>0) return false;
Arrays.parallelSort(hand);
Map<Integer,Integer> map=new HashMap<>();
for(int i=0;i<n;i++)
map.put(hand[i], map.getOrDefault(hand[i], 0)+1);
for(int i=0;i<n;i++) {
if(map.get(hand[i])<=0)
continue;
for(int j=hand[i];j<hand[i]+W;j++) {
if(!map.containsKey(j) || map.get(j)<=0)
return false;
map.put(j, map.get(j)-1);
}
}
return true;
}
}