LeetCode每日一题[C++]-猜数字游戏

题目描述

链接：. - 力扣（LeetCode）

你在和朋友一起玩猜数字（Bulls and Cows）游戏，该游戏规则如下：

写出一个秘密数字，并请朋友猜这个数字是多少。朋友每猜测一次，你就会给他一个包含下述信息的提示：

猜测数字中有多少位属于数字和确切位置都猜对了（称为 "Bulls"，公牛），
有多少位属于数字猜对了但是位置不对（称为 "Cows"，奶牛）。也就是说，这次猜测中有多少位非公牛数字可以通过重新排列转换成公牛数字。

给你一个秘密数字 secret 和朋友猜测的数字 guess ，请你返回对朋友这次猜测的提示。

提示的格式为 "xAyB" ，x 是公牛个数， y 是奶牛个数，A 表示公牛，B 表示奶牛。

请注意秘密数字和朋友猜测的数字都可能含有重复数字。

示例 1：

复制代码

输入：secret = "1807", guess = "7810"
输出："1A3B"
解释：数字和位置都对（公牛）用 '|' 连接，数字猜对位置不对（奶牛）的采用斜体加粗标识。
"1807"
  |
"7810"

示例 2：

复制代码

输入：secret = "1123", guess = "0111"
输出："1A1B"
解释：数字和位置都对（公牛）用 '|' 连接，数字猜对位置不对（奶牛）的采用斜体加粗标识。
"1123"        "1123"
  |      or     |
"0111"        "0111"
注意，两个不匹配的 1 中，只有一个会算作奶牛（数字猜对位置不对）。通过重新排列非公牛数字，其中仅有一个 1 可以成为公牛数字。

解题思路

我们创建两个计数器 cnt1和 cnt2，分别用来统计秘密数字和朋友猜测的数字中每个数字出现的次数。同时，我们创建一个变量 x 来统计公牛的数量。

然后我们遍历秘密数字和朋友猜测的数字，如果当前数字相同，我们就将 x加一。否则，我们分别将秘密数字和朋友猜测的数字中的当前数字的计数加一。

最后，我们遍历 cnt1中的每个数字，取 cnt1 和 cnt2中当前数字的计数的最小值，然后将这个最小值加到变量 y 中。

最后，我们返回 x 和 y 的值。

复杂度分析

时间复杂度 O(n)，其中 n 是秘密数字和朋友猜测的数字的长度。

空间复杂度 O(∣Σ∣)，其中 ∣Σ∣ 是字符集的大小，本题中字符集为数字，所以 ∣Σ∣=10。

代码

c++

cpp 复制代码

#include <vector>
#include <iostream>
using namespace std;
class Solution
{
public:
    string getHint(string secret, string guess)
    {
        int x = 0, y = 0;
        int cnt1[10]{};
        int cnt2[10]{};
        for (int i = 0; i < secret.size(); ++i)
        {
            int a = secret[i] - '0', b = guess[i] - '0';
            if (a == b)
            {
                ++x;
            }
            else
            {
                ++cnt1[a];
                ++cnt2[b];
            }
        }
        for (int i = 0; i < 10; ++i)
        {
            y += min(cnt1[i], cnt2[i]);
        }
        return to_string(x) + "A" + to_string(y) + "B";
    }
};
int main()
{
    Solution solution;
    string secret = "1807";
    string guess = "7810";
    string res = solution.getHint(secret, guess);
    cout << "res: " << res << endl;

    secret = "1123";
    guess = "0111";
    res = solution.getHint(secret, guess);
    cout << "res: " << res << endl;
}

python：

python 复制代码

from collections import Counter


class Solution:
    def getHint(self, secret: str, guess: str) -> str:
        cnt1, cnt2 = Counter(), Counter()
        x = 0
        for a, b in zip(secret, guess):
            if a == b:
                x += 1
            else:
                cnt1[a] += 1
                cnt2[b] += 1
        y = sum(min(cnt1[c], cnt2[c]) for c in cnt1)
        return f"{x}A{y}B"


if __name__ == '__main__':
    secret = "1807"
    guess = "7810"
    print(Solution().getHint(secret, guess))
    secret = "1123"
    guess = "0111"
    print(Solution().getHint(secret, guess))