[编程题]雀魂启动

链接：雀魂启动！_字节跳动笔试题_牛客网
来源：牛客网
 

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

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

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

示例1

输入

1 1 1 2 2 2 5 5 5 6 6 6 9

输出

9

说明

可以组成1,2,6,7的4个刻子和9的雀头

示例2

输入

1 1 1 1 2 2 3 3 5 6 7 8 9

输出

4 7

说明

用1做雀头，组123,123,567或456,789的四个顺子

示例3

输入

1 1 1 2 2 2 3 3 3 5 7 7 9

输出

0

说明

来任何牌都无法和牌

备注:

请不要根据自己的常识为题目增加未提及的条件

对于20%的数据，保证和牌时一定是4个刻子+雀头的形式

import java.util.*;

public class Main {
    public static List<Integer> solve(int[] cntArr) {
        List<Integer> ans = new ArrayList<>();
        for (int i = 1; i < 10; i++) {
            if (cntArr[i] < 4) {
                int[] tmpArr = Arrays.copyOf(cntArr, cntArr.length);
                tmpArr[i] += 1;
                if (isDz(tmpArr)) {
                    ans.add(i);
                }
                tmpArr[i] -= 1;
            }
        }
        return ans;
    }

    public static boolean isDz(int[] cntArr) {
        // 对子
        for (int i = 1; i < 10; i++) {
            if (cntArr[i] > 1) {
                cntArr[i] -= 2;
                if (isHu(cntArr, 12, 4)) {
                    return true;
                }
                cntArr[i] += 2;
            }
        }
        return false;
    }

    public static boolean isHu(int[] cntArr, int count, int n) {
        if (count == 0 && n == 0) {
            return true;
        }
        for (int i = 1; i < 10; i++) {
            if (cntArr[i] > 2) {
                cntArr[i] -= 3;
                if (isHu(cntArr, count - 3, n - 1)) {
                    return true;
                }
                cntArr[i] += 3;
            }
        }
        for (int i = 1; i < 8; i++) {
            if (cntArr[i] > 0 && cntArr[i + 1] > 0 && cntArr[i + 2] > 0) {
                cntArr[i] -= 1;
                cntArr[i + 1] -= 1;
                cntArr[i + 2] -= 1;
                if (isHu(cntArr, count - 3, n - 1)) {
                    return true;
                }
                cntArr[i] += 1;
                cntArr[i + 1] += 1;
                cntArr[i + 2] += 1;
            }
        }
        return false;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int[] cntArr = new int[10];
        int tmp;
        for (int i = 0; i < 13; i++) {
            tmp = sc.nextInt();
            cntArr[tmp]++;
        }
        List<Integer> ans = solve(cntArr);
        if (ans.size() < 1) {
            System.out.println(0);
        } else {
            System.out.print(ans.get(0));
            for (int i = 1; i < ans.size(); i++) {
                System.out.print(" ");
                System.out.print(ans.get(i));
            }
        }
    }
}