华为OD机考——斗地主之找顺子

题目描述

在斗地主只扑克牌游戏中，扑克牌由小到大的顺序为:3.4,5.6,7.8,9,10.J,Q.K.A.2，玩家可以出的扑克牌阵型有:单张、对子、顺子、飞机、炸弹等。

其中顺子的出牌规则为:由至少5张由小到大连续递增的扑克牌只组成，且不能包含2。

例如:(3.4,5,6,7}、(3.4,5,6,7,8,9,10,J,Q,K,A}都是有效的顺子;而{J,Q,K,A,2}、(2,3,4,5,6}、(3,4,5,6}、(3,4,5.6,8)等都不是顺子给定一个包含 13 张牌的数组，如果有满足出牌规则的顺子，请输出顺子。

如果存在多个顺子，请每行输出一个顺子，且需要按顺子的第一张牌的大小(必须从小到大)依次输出。

如果没有满足出牌规则的顺子，请输出NO。

输入描述

13张任意顺序的扑克牌，每张扑克牌数字用空格隔开，每张扑克牌的数字都是合法的，并且不包括大小王:2 9 J 2 3 4 K A 7 9 A 5 6不需要考虑输入为异常字符的情况

输出描述

组成的顺子，每张扑克牌数字用空格隔开:3 4 5 6 7

示例1

输入

2 9 J 2 3 4 K A 7 9 A 5 6

输出

3 4 5 6 7

说明

13张牌中，可以组成的顺子只有1组:3 4 5 6 7.

示例2

输入

2 9 J 10 3 4 K A 7 Q A 5 6

输出

3 4 5 6 79 10 J Q K A

说明

13张牌中，可以组成2组顺子，从小到大分别为:3 4 5 6 7和9 10 J Q K A

示例3

输入

2 9 9 9 3 4 K A 10 Q A 5 6

输出

NO

说明

13张牌中，无法组成顺子。

我的解法

const readline = require('readline');

const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
});

let bucket = new Array(13).fill(0);
let sortedCards = [];
rl.on('line', (line) => {
    // 初始化牌堆桶
    initBucketAndCards(line);

    const isStraight = checkStraight();
    if (isStraight) {
        const sequences = findSequences();
        sequences.forEach(seq => console.log(seq.join(' ')));
    } else {
        console.log('NO');
    }
    rl.close();
});

function initBucketAndCards(line) {
    const cards = line.split(' ').map(card => {
        if (card === 'J') return 11;
        if (card === 'Q') return 12;
        if (card === 'K') return 13;
        if (card === 'A') return 14;
        return parseInt(card);
    }).filter(card => card >= 3 && card <= 14);
    //初始化排序手牌
    sortedCards = cards.sort((a, b) => a - b);
    //初始化卡牌桶
    for (const card of sortedCards) {
        bucket[card - 3]++;
    }
}

//这里只需要看是否有连续大于五张牌就可以了
function checkStraight() {
    let continuousCount = 0;
    for (const count of bucket) {
        if (count > 0) {
            continuousCount++;
            if (continuousCount >= 5) return true;
        } else {
            continuousCount = 0;
        }
    }
    return false;
}

//找到连续的顺子
function findSequences() {
    const sequences = [];

    while (checkStraight()) {
        const seq = [];
        for (let i = 0; i < 13; i++) {
            if (bucket[i] > 0) {
                seq.push(i + 3);
                bucket[i]--;
            } else if (bucket[i] < 1 && seq.length >= 5) {
                sequences.push([...seq]);
                break;
            } else {
                break;
            }
        }
    }
    return sequences.map(seq => seq.map(card => card === 11 ? 'J' : card === 12 ? 'Q' : card === 13 ? 'K'
        : card === 14 ? 'A' : card.toString()));
}