愛麗絲有一手(hand)由整數(shù)數(shù)組給定的牌。
現(xiàn)在她想把牌重新排列成組,使得每個組的大小都是 W,且由 W 張連續(xù)的牌組成。
如果她可以完成分組就返回 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 的組。
思路
使用TreeMap表示每張牌和出現(xiàn)的次數(shù),每次循環(huán)W次比較排是否連續(xù),遍歷過得出現(xiàn)次數(shù) - 1,如果減為0以后從map里移除。
public boolean isNStraightHand(int[] hand, int W) {
TreeMap<Integer,Integer> map = new TreeMap<>();
for (int i = 0; i < hand.length; i++) {
if (map.containsKey(hand[i])) {
map.put(hand[i],map.get(hand[i]) + 1);
}else {
map.put(hand[i],1);
}
}
while (map.size() > 0) {
int first = map.firstKey(); //因為treemap是有序的,第一個key就是最小的key,如果用hashmap還要去找最小值
for (int i = 0; i < W; i++) {
if (map.containsKey(first + i)) {
if (map.get(first + i) - 1 == 0) {
map.remove(first + i);
} else {
map.put(first + i, map.get(first + i) - 1);
}
} else {
return false;
}
}
}
return true;
}