题目
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 的组。
解题
方法一:贪心
首先从小到大排序
通过map记录每个数出现的次数。
然后遍历hand, 如果存在x,就map[x]-- ,然后找x+1,找到就map[x+1]–,…否则的话,就说明找不到,返回false
也就是说按照遇到的第一个数,当作顺子的开头,然后寻找第2、3、…、groupSize的数是否存在,不存在就返回false。
class Solution { public: bool isNStraightHand(vector<int>& hand, int groupSize) { int n=hand.size(); if(n%groupSize!=0) return false; sort(hand.begin(),hand.end()); unordered_map<int,int> map; for(int x:hand){ map[x]++; } for(int x:hand){ if(map.count(x)==0) continue; for(int j=0;j<groupSize;j++){ int num=x+j; if(map.count(num)==0) return false; map[num]--; if(map[num]==0) map.erase(num); } } return true; } };