题目描述(中等)
Alice 手中有一把牌,她想要重新排列这些牌,分成若干组,使每一组的牌数都是 groupSize ,并且由 groupSize 张连续的牌组成。
给你一个整数数组 hand 其中 hand[i] 是写在第 i 张牌,和一个整数 groupSize 。如果她可能重新排列这些牌,返回 true ;否则,返回 false 。
示例 1:
输入:hand = [1,2,3,6,2,3,4,7,8], groupSize = 3
输出:true
解释:Alice 手中的牌可以被重新排列为 [1,2,3],[2,3,4],[6,7,8]。
示例 2:
输入:hand = [1,2,3,4,5], groupSize = 4
输出:false
解释:Alice 手中的牌无法被重新排列成几个大小为 4 的组。
提示:
1 <= hand.length <= 10^4
0 <= hand[i] <= 10^9
1 <= groupSize <= hand.length
思路
首先判断边界情况,如果顺子大小是1直接返回true,如果牌数不能整除顺子大小,返回false;
开始用哈希表计数,统计每个数出现的次数;
之后用一个数组存所有出现过的数字,并对其从小到大排序;
因为顺子必然连续,贪心的从小的开始,则必然是一个顺子的开头,如果它出现了nsize次,则从它之后顺子大小长度的数都必须出现大于等于这个次数,如果满足,则减去这个次数,相当于打出顺子了,如果不满足就false;
如果全能打完就返回true。
代码
class Solution {
public:
bool isNStraightHand(vector<int>& hand, int gs) {
int n = hand.size();
if(gs == 1) return true;
if(n%gs != 0) return false;
unordered_map<int, int> count;
vector<int> nums;
for(auto x : hand)
count[x]++;
for(auto x : count)
nums.push_back(x.first);
sort(nums.begin(), nums.end());
for(auto num : nums) {
int nsize = count[num];
if(nsize > 0) {
for(int i = num; i < num + gs; i++) {
if(count[i] < nsize) return false;
count[i] -= nsize;
}
}
}
return true;
}
};