一.题目描述
小包最近迷上了一款叫做雀魂的麻将游戏,但是这个游戏规则太复杂,小包玩了几个月了还是输多赢少。
于是生气的小包根据游戏简化了一下规则发明了一种新的麻将,只留下一种花色,并且去除了一些特殊和牌方式(例如七对子等),具体的规则如下:
https://www.nowcoder.com/questionTerminal/448127caa21e462f9c9755589a8f2416
二.代码
public ArrayList<Integer> a(int[] input) {
int[] state = new int[9], helpArr = new int[9];
ArrayList<Integer> res = new ArrayList<>();
for (int i = 0; i < input.length; i++) {
state[input[i] - 1]++;
}
for (int i = 0; i < 9; i++) {
if (state[i] < 4) {
int num = i + 1;
System.arraycopy(state, 0, helpArr, 0, 9);
helpArr[i]++;
if (canHu(helpArr, 14, false)) res.add(num);
}
}
return res;
}
private boolean canHu(int[] arr, int total, boolean hasHead) {
if (total == 0) return true;
if (!hasHead) {
for (int i = 0; i < 9; i++) {
if (arr[i] >= 2) {
arr[i] -= 2;
if (canHu(arr, total - 2, true)) return true;
arr[i] += 2;
}
}
return false;
} else {
for (int i = 0; i < 9; i++) {
if (arr[i] > 0) {
if (arr[i] >= 3) {
arr[i] -= 3;
if (canHu(arr, total - 3, true)) return true;
arr[i] += 3;
}
if (i + 2 < 9 && arr[i + 1] > 0 && arr[i + 2] > 0) {
arr[i]--;
arr[i + 1]--;
arr[i + 2]--;
if (canHu(arr, total - 3, true)) return true;
arr[i]++;
arr[i + 1]++;
arr[i + 2]++;
}
}
}
}
return false;
}
}