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 <= 104 0 <= hand[i] <= 109 1 <= groupSize <= hand.length

题解: 一开始就想到了贪心的思路,也想到了哈希,但是没有想到将他们结合在一起,结果就写了一个复杂度特别高的代码,我复杂度应该是有O(n2logn)了,实际上复杂度应该是O(nlogn)为最佳的。

//O(n2logn)复杂度版本,每次剔除都是从头遍历。
class Solution {
public:
    bool isNStraightHand(vector<int>& hand, int groupSize) {
        sort(hand.begin(), hand.end());
        int n = hand.size();
        if(n % groupSize != 0) return false;
        vector<bool> vis(n);
        int i = 0, j = 0;
        for(int i = 0; i < n; i++){
            if(!vis[i]){
                vis[i] = true;
                j = i + 1;
                int cnt = 1;
                int tmp = hand[i];
                while(j < n && cnt < groupSize){
                    if(!vis[j] && hand[j] == tmp + 1){
                        tmp = hand[j];
                        cnt++;
                        vis[j] = true;
                    }
                    j++;
                }
                if(cnt < groupSize) return false;
            }
        }
        for(int i = 0; i < n; i++){
            if(!vis[i]){
                return false;
            }
        }
        return true;
    }
};

下面附上O(nlogn)版本的复杂度。

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> cnt;
        for (auto & num : hand) {
            cnt[num]++;
        }
        for (auto & x : hand) {
            if (!cnt.count(x)) {
                continue;
            }
            for (int j = 0; j < groupSize; j++) {
                int num = x + j;
                if (!cnt.count(num)) {
                    return false;
                }
                cnt[num]--;
                if (cnt[num] == 0) {
                    cnt.erase(num);
                }
            }
        }
        return true;
    }
};