这个编程题是来自于牛客网的字节跳动2019春招研发部分编程题汇总,有感兴趣的朋友们可以去原网站查看。
小包最近迷上了一款叫做雀魂的麻将游戏,但是这个游戏规则太复杂,小包玩了几个月了还是输多赢少。 于是生气的小包根据游戏简化了一下规则发明了一种新的麻将,只留下一种花色,并且去除了一些特殊和牌方式(例如七对子等),具体的规则如下:
- 总共有36张牌,每张牌是1~9。每个数字4张牌。
- 你手里有其中的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个刻子+雀头的形式
解析
本题采用递归中的回溯,每一次回溯都从当前剩下的牌中去掉一种牌型,若到最后所剩的牌为空,则代表能够和牌,因此就将这张牌加到输出里
C++源码
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
bool isHu(vector<int> vec) {
if (vec.size() == 0)
return true;
int len = vec.size();
int first = vec[0];
int count0 = count(vec.begin(), vec.end(), first);
if ((len % 3 != 0) && count0 >= 2 && isHu(vector<int>(vec.begin()+2, vec.end())))
return true;
if (count0 >= 3 && isHu(vector<int>(vec.begin()+3, vec.end())))
return true;
auto it1 = find(vec.begin(), vec.end(), first+1);
auto it2 = find(vec.begin(), vec.end(), first+2);
if ((it1 != vec.end()) && (it2 != vec.end())) {
vec.erase(vec.begin(), vec.begin()+1);
it1 = find(vec.begin(), vec.end(), first+1);
vec.erase(it1, it1+1);
it2 = find(vec.begin(), vec.end(), first+2);
vec.erase(it2, it2+1);
if (isHu(vec))
return true;
}
return false;
}
int main() {
vector<int> sVec(13);
for (size_t i = 0; i < sVec.size(); ++i)
cin >> sVec[i];
map<int, int> sMap;
for (size_t i = 0; i < sVec.size(); ++i) {
if (sMap.find(sVec[i]) != sMap.end())
sMap[sVec[i]]++;
else sMap[sVec[i]] = 1;
}
vector<int> result;
for (int i = 1; i < 10; ++i) {
if (sMap.find(i) != sMap.end() && sMap[i] == 4)
continue;
vector<int> vec(sVec);
vec.push_back(i);
sort(vec.begin(), vec.end());
if (isHu(vec))
result.push_back(i);
}
if (result.empty())
result.push_back(0);
for (size_t i = 0; i < result.size(); ++i)
cout << result[i] << " ";
return 0;
}