一手顺子 | 徐年の博客

力扣 846. 一手顺子

题目说明

Alice 手中有一把牌，她想要重新排列这些牌，分成若干组，使每一组的牌数都是 groupSize ，并且由 groupSize 张连续的牌组成。

给你一个整数数组 hand 其中 hand[i] 是写在第 i 张牌，和一个整数 groupSize 。如果她可能重新排列这些牌，返回 true ；否则，返回 false 。

提示：

1 <= hand.length <= 10^4
0 <= hand[i] <= 10^9
1 <= groupSize <= hand.length

示例

示例 1:

1
2
3

输入：hand = [1,2,3,6,2,3,4,7,8], groupSize = 3
输出：true
解释：Alice 手中的牌可以被重新排列为 [1,2,3]，[2,3,4]，[6,7,8]。

示例 2:

1
2
3

输入：hand = [1,2,3,4,5], groupSize = 4
输出：false
解释：Alice 手中的牌无法被重新排列成几个大小为 4 的组。

笔者理解

此题是一道数组算法问题，在力扣题库中被定义为中等题。

解法

当笔者阅读完此题后，发现此题可以采用贪心策略，我们将所有数据存进优先队列，如何取一个最小的数，然后再在优先队列中将后面的序列依次筛查即可，让我们来看看具体如何实现的吧。

实现

class Solution {
    /**
     * 优先队列
     * 贪心
     */
    public boolean isNStraightHand(int[] hand, int groupSize) {
        int n = hand.length;
        // 特例
        if (groupSize == 1) {
            return true;
        }

        PriorityQueue<Integer> queue = new PriorityQueue<>();

        for (int h: hand) {
            queue.add(h);
        }

        while (!queue.isEmpty()) {
            // 取出一个元素作为首元素
            int first = queue.poll();

            // 序列删除，如果队列中不包含后序元素则直接返回 false
            for (int i = 1; i < groupSize; i++) {
                if (!queue.remove(first + i)) {
                    return false;
                }
            }
        }

        // 所有元素排完
        return true;
    }
}