校招真题题解:雀魂启动!

2023/07/11 leetcode 算法 共 2105 字,约 7 分钟

字节跳动春招题,不知道是哪一年的。原题如下:

雀魂启动!

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

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

  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张牌中,再取一张牌,取到哪几种数字牌可以和牌。

时间限制: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")

文档信息

Search

    Table of Contents