字节跳动春招题,不知道是哪一年的。原题如下:
雀魂启动!
小包最近迷上了一款叫做雀魂的麻将游戏,但是这个游戏规则太复杂,小包玩了几个月了还是输多赢少。
于是生气的小包根据游戏简化了一下规则发明了一种新的麻将,只留下一种花色,并且去除了一些特殊和牌方式(例如七对子等),具体的规则如下:
- 总共有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张牌中,再取一张牌,取到哪几种数字牌可以和牌。
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32M,其他语言64M
输入描述:
输入只有一行,包含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
例子说明:
来任何牌都无法和牌
这道题如果直接模拟条件会非常复杂,自然的思路就是用回溯的方法进行选择。也可以看作dfs,只是对于这种回溯只要有一个路径满足要求就可以return true。这种类似的题目都可以采用这样类似的思路:
#include <iostream>
#include<vector>
#include<algorithm>
using namespace std;
// 判断是否满足和牌,使用回溯的思想
void tell(vector<int>& nums, int quetou, int shunzi, int cnt, bool& tag){
if(cnt==14&&quetou==1&&shunzi==4){
tag=true;
return;
}
if(quetou==0){
for(int i=1;i<=9;++i){
if(nums[i]>=2){
nums[i]-=2;
cnt+=2;
quetou++;
tell(nums, quetou, shunzi, cnt, tag);
nums[i]+=2;
cnt-=2;
quetou--;
}
}
}
else{
for(int i=1;i<=9;++i){
if(nums[i]>=3){
nums[i]-=3;
cnt+=3;
shunzi++;
tell(nums,quetou,shunzi,cnt, tag);
nums[i]+=3;
cnt-=3;
shunzi--;
}
else if(nums[i]>=1){
if(i+2<=9&&nums[i+1]>=1&&nums[i+2]>=1){
nums[i]--;
nums[i+1]--;
nums[i+2]--;
shunzi++;
cnt+=3;
tell(nums,quetou,shunzi,cnt, tag);
nums[i]++;
nums[i+1]++;
nums[i+2]++;
cnt-=3;
shunzi--;
}
}
}
}
}
int main() {
vector<int> nums(10,0);
for(int i=0;i<13;++i){
int x;
cin>>x;
nums[x]++;
}
vector<int> ans;
for(int i=1;i<=9;++i){
if(nums[i]<4){
// 暴力求解,一共只可能有1-9,都试一遍能不能满足条件
nums[i]++;
bool tag=false;
tell(nums,0,0,0,tag);
if(tag){
ans.push_back(i);
}
nums[i]--;
}
}
if(ans.size()==0){
cout<<0<<endl;
}
else{
sort(ans.begin(), ans.end());
cout<<ans[0];
for(int i=1;i<ans.size();++i){
cout<<" "<<ans[i];
}
}
}
// 64 位输出请用 printf("%lld")