字节跳动——麻将和牌

原题描述：

雀魂启动！

小包最近迷上了一款叫做雀魂的麻将游戏，但是这个游戏规则太复杂，小包玩了几个月了还是输多赢少。

于是生气的小包根据游戏简化了一下规则发明了一种新的麻将，只留下一种花色，并且去除了一些特殊和牌方式（例如七对子等），具体的规则如下：

1. 总共有36张牌，每张牌是1~9。每个数字4张牌。

2. 你手里有其中的14张牌，如果这14张牌满足如下条件，即算作和牌

• 14张牌中有2张相同数字的牌，称为雀头。
• 除去上述2张牌，剩下12张牌可以组成4个顺子或刻子。顺子的意思是递增的连续3个数字牌（例如234,567等），刻子的意思是相同数字的3个数字牌（例如111,777)

例如：

1 1 1 2 2 2 6 6 6 7 7 7 9 9 可以组成1,2,6,7的4个刻子和9的雀头，可以和牌
1 1 1 1 2 2 3 3 5 6 7 7 8 9 用1做雀头，组123,123,567,789的四个顺子，可以和牌
1 1 1 2 2 2 3 3 3 5 6 7 7 9无论用1 2 3 7哪个做雀头，都无法组成和牌的条件。

现在，小包从36张牌中抽取了13张牌，他想知道在剩下的23张牌中，再取一张牌，取到哪几种数字牌可以和牌。 

输入描述：

输入只有一行，包含13个数字，用空格分隔，每个数字在1~9之间，数据保证同种数字最多出现4次。

输出描述：

输出同样是一行，包含1个或以上的数字。代表他再取到哪些牌可以和牌。若满足条件的有多种牌，请按从小到大的顺序输出。若没有满足条件的牌，请输出一个数字0

题目精炼：

现需要用14张牌（包含1-9的数字，每种数字最多4张）来和牌。和牌条件是：有2张相同的牌作为雀头，剩下的12张牌能组成4个顺子（3张连续递增）或刻子（3张相同）。现在有13张牌，想知道再摸哪张牌能和牌，输出能和牌的数字，按从小到大排序，如果没有则输出0。

解题思路：

将问题逐一拆解，并用数组模拟哈希表：下标为key，表示数字；值为value，表示对应牌的张数

1）由于此时只有13张牌，因此我们首先通过循环遍历插入新的牌。如果该牌的数量已经达到4张，则返回null，表示无法再添加该牌；否则，增加该牌的数量并返回更新后的cards数组。

2）当此时有14张牌，循环遍历选择雀头，检查指定的牌是否可以作为雀头（即该牌至少有两张）。如果可以，cards[i] -= 2，并调用checkHu方法，检查剩余的12张牌是否满足和牌的条件。

3）检查这12张牌是否能组成4个顺子或刻子，如果能，返回true，插入的牌可以和牌。

• 4 张牌 --> 选3张牌组成刻子 --> 1张牌 --> 接下来处理1张牌的情况
• 1张牌 --> 组成一对顺子
• 3张牌 --> 组成刻子
• 2张牌 --> 组成两对顺子

注意：每次递归检查，要通过复制数组来避免在检查过程中修改原始牌组，保证每次检查都是基于原始状态。 

import java.util.Arrays;
import java.util.Scanner;
public class Mahjong {
    static int[] init(int[] nums){
        int[] cards = new int[10];
        for (int num : nums) {
            cards[num] += 1;
        }
        return cards;
    }
    static int[] addNewCard(int newCard, int[] cards){
        if(cards[newCard] >= 4){
            return null;
        }else{
            cards[newCard] += 1;
            return cards;
        }
    }
    static boolean isHead(int head, int[] cards){
        if(cards[head] < 2){
            return false;
        }else{
            cards[head] -= 2;
        }
        return checkHu(cards);
    }


    static boolean checkHu( int[] cards) {
        int count = 0;
        for (int i = 1; i < cards.length; i++) {
            if (cards[i] == 0) continue;
            if (cards[i] == 4) {
                cards[i] -= 3;
                count++;
            }
        }
        for (int i = 1; i < cards.length - 2; i++) {
            if (cards[i] == 0) continue;
            if (cards[i] == 1) {
                if (cards[i + 1] >= 1 && cards[i + 2] >= 1) {
                    cards[i] -= 1;
                    cards[i + 1] -= 1;
                    cards[i + 2] -= 1;
                    count++;
                } else {
                    return false;
                }
            }
        }
        if (count == 4) return true;
        for (int i = 1; i < cards.length; i++) {
            if (cards[i] == 0) continue;
            if (cards[i] == 3) {
                cards[i] -= 3;
                count++;
            }
        }
        if (count == 4) return true;

        for (int i = 1; i < cards.length - 2; i++) {
            if (cards[i] == 0) continue;
            if (cards[i] == 2) {
                if (cards[i + 1] == 2 && cards[i + 2] == 2) {
                    cards[i] -= 2;
                    cards[i + 1] -= 2;
                    cards[i + 2] -= 2;
                    count += 2;
                } else {
                    return false;
                }
            }
        }
        return count == 4;
    }



    public static void main(String[] args) {
        int[] nums = new int[13];
        Scanner in = new Scanner(System.in);
        for (int i = 0; i < nums.length; i++) {
            nums[i] = in.nextInt();
        }
        int[] initialCards = init(nums);
        boolean flag = false;
        for (int i = 1; i <= 9 ; i++) {
            int[] cards = addNewCard(i,Arrays.copyOf(initialCards, initialCards.length));
            for (int j = 1; j <= 9; j++) {
                if (cards != null) {
                    if (isHead(j, Arrays.copyOf(cards, cards.length))) {
                        flag = true;
                        System.out.print(i+" ");
                    }
                }else{
                    break;
                }
            }
        }
        if(!flag){
            System.out.print(0);
        }
    }
}