爱丽丝有一手(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
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/hand-of-straights 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
bool isNStraightHand(vector<int>& hand, int W) {
int i, n = hand.size(), prev, count = 0;
if(n%W) return false;//个数不能整除,错
if(W == 1) return true;//1肯定对
map<int,int> m;//数字,个数
for(i = 0; i < n; ++i)
m[hand[i]]++;
auto it = m.begin();
for(i = 0; i < n; ++i)
{
if(count != 0 && it->first - prev != 1)
return false;//比前一个多1吗
prev = it->first;
m[prev]--;//计数-1
if(m[prev] == 0)
m.erase(it++);//计数为0,删除,这么写,避免迭代器失效
else
it++;
if(it == m.end())//到末尾了,跳到begin
it = m.begin();
count++;//个数+1
if(count == W)
{ //满了W个
count = 0;
it = m.begin();
}
}
return true;
}
};
168 ms 15.9 MB