哈希表题目：猜数字游戏

题目

标题和出处

标题：猜数字游戏

出处：299. 猜数字游戏

难度

4 级

题目描述

要求

你在和朋友一起玩猜数字（Bulls and Cows）游戏。

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

• 「公牛」的数量，即猜测数字中有多少位数字在秘密数字中出现且位置正确。
• 「奶牛」的数量，即猜测数字中有多少位数字在秘密数字中出现但位置错误，即猜测数字中有多少位非公牛数字可以通过重新排列变成公牛数字。

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

提示的格式为 $\text{"xAyB"}$，$\text{x}$ 是公牛个数，$\text{y}$ 是奶牛个数。请注意 $\text{secret}$ 和 $\text{guess}$ 都可能含有重复数字。

示例

示例 1：

输入：$\text{secret = "1807", guess = "7810"}$
输出：$\text{"1A3B"}$
解释：数字和位置都对（公牛）用 $\text{‘|’}$ 连接，数字猜对位置不对（奶牛）用下划线表示。
$\text{"1807"}$
$\text{ |}$
$\text{"}\underline{\text{7}}\text{8}\underline{\text{10}}\text{"}$

示例 2：

输入：$\text{secret = "1123", guess = "0111"}$
输出：$\text{"1A1B"}$
解释：数字和位置都对（公牛）用 $\text{‘|’}$ 连接，数字猜对位置不对（奶牛）用下划线表示。
$\text{"1123"}$
$\text{ |}$
$\text{"01}\underline{\text{1}}\text{1"}$
或
$\text{"1123"}$
$\text{ |}$
$\text{"011}\underline{\text{1}}\text{"}$
注意，两个不匹配的 $\text{1}$ 中，只有一个会算作奶牛（数字猜对位置不对），因为通过重新排列非公牛数字之后只有一个 $\text{1}$ 可以成为公牛数字。

示例 3：

输入：$\text{secret = "1", guess = "0"}$
输出：$\text{"0A0B"}$

示例 4：

输入：$\text{secret = "1", guess = "1"}$
输出：$\text{"1A0B"}$

数据范围

• $\text{1}\le \text{secret.length, guess.length}\le \text{1000}$
• $\text{secret.length = guess.length}$
• $\text{secret}$ 和 $\text{guess}$ 仅由数字组成

解法一

思路和算法

公牛为 $\text{secret}$ 和 $\text{guess}$ 的相同下标处数字相同的下标个数，统计公牛的方法是：同时遍历 $\text{secret}$ 和 $\text{guess}$，对于每个下标，如果 $\text{secret}$ 和 $\text{guess}$ 在相同下标处的数字相同，则将公牛的数量加 $1$。

奶牛为 $\text{secret}$ 和 $\text{guess}$ 的不同下标处数字相同的下标个数，统计奶牛的方法是：使用两个哈希表分别记录 $\text{secret}$ 和 $\text{guess}$ 中的每个数字的出现次数（排除公牛的情况），同时遍历 $\text{secret}$ 和 $\text{guess}$，对于每个下标，如果 $\text{secret}$ 和 $\text{guess}$ 在相同下标处的数字不同，则将该下标处的两个数字在对应的哈希表中的出现次数分别加 $1$，遍历结束之后，$0$ 到 $9$ 的每个数字在奶牛中的数量为该数字在两个哈希表中的出现次数的较小值，计算每个数字在奶牛中的数量之和，即为奶牛的数量。

由于统计公牛和奶牛都需要同时遍历 $\text{secret}$ 和 $\text{guess}$，因此可以在一次遍历中完成统计公牛和奶牛的操作。遍历过程中，如果 $\text{secret}$ 和 $\text{guess}$ 在相同下标处的数字相同则将公牛的数量加 $1$，如果 $\text{secret}$ 和 $\text{guess}$ 在相同下标处的数字不同则将该下标处的两个数字在 $\text{secret}$ 和 $\text{guess}$ 中的出现次数分别加 $1$，遍历结束之后遍历 $0$ 到 $9$ 在 $\text{secret}$ 和 $\text{guess}$ 中的出现次数，统计奶牛的数量。

实现方面，由于数字范围是 $0$ 到 $9$，因此可以使用两个数组分别记录 $\text{secret}$ 和 $\text{guess}$ 中的每个数字的出现次数。

代码

class Solution {
    public String getHint(String secret, String guess) {
        int bulls = 0, cows = 0;
        int[] secretCounts = new int[10];
        int[] guessCounts = new int[10];
        int length = secret.length();
        for (int i = 0; i < length; i++) {
            int secretDigit = secret.charAt(i) - '0';
            int guessDigit = guess.charAt(i) - '0';
            if (secretDigit == guessDigit) {
                bulls++;
            } else {
                secretCounts[secretDigit]++;
                guessCounts[guessDigit]++;
            }
        }
        for (int i = 0; i < 10; i++) {
            cows += Math.min(secretCounts[i], guessCounts[i]);
        }
        return String.format("%dA%dB", bulls, cows);
    }
}

复杂度分析

• 时间复杂度：$O(n+C)$，其中 $n$ 是字符串 $\text{secret}$ 和 $\text{guess}$ 的长度，$C$ 是数字的个数，$C=10$。需要 $O(n)$ 的时间遍历字符串 $\text{secret}$ 和 $\text{guess}$ 各一次统计公牛数量和每个数字的出现次数，需要 $O(C)$ 的时间遍历 $0$ 到 $9$ 的所有数字统计奶牛数量。

• 空间复杂度：$O(C)$，其中 $C$ 是数字的个数，$C=10$。需要使用两个哈希表分别记录 $\text{secret}$ 和 $\text{guess}$ 中的每个数字的出现次数。

解法二

思路和算法

解法一需要使用两个哈希表分别记录 $\text{secret}$ 和 $\text{guess}$ 中的每个数字的出现次数，在遍历 $\text{secret}$ 和 $\text{guess}$ 之后还需要遍历 $0$ 到 $9$ 的每个数字计算奶牛的数量。可以只用一个哈希表，在遍历 $\text{secret}$ 和 $\text{guess}$ 的同时计算奶牛的数量。

使用一个哈希表记录每个数字的出现次数（排除公牛的情况），遍历 $\text{secret}$ 和 $\text{guess}$ 的过程中，对于 $\text{secret}$ 中出现的每个数字将出现次数加 $1$，对于 $\text{guess}$ 中出现的每个数字将出现次数减 $1$。上述操作的思想是：对于 $\text{secret}$ 中的每个数字，只有当 $\text{guess}$ 中存在一个不同位置的相同数字时，才是一个奶牛，反之亦然。

当遍历到一个下标时，将 $\text{secret}$ 和 $\text{guess}$ 在该下标处的数字分别记为 $\text{secretDigit}$ 和 $\text{guessDigit}$，如果 $\text{secretDigit}=\text{guessDigit}$ 则将公牛的数量加 $1$，如果 $\text{secretDigit}\ne \text{guessDigit}$ 则按照如下操作统计奶牛：

1. 如果哈希表中 $\text{secretDigit}$ 的出现次数小于 $0$，则在之前遍历的 $\text{guess}$ 的数字中有多余的 $\text{secretDigit}$，因此将奶牛的数量加 $1$；

2. 如果哈希表中 $\text{guessDigit}$ 的出现次数大于 $0$，则在之前遍历的 $\text{secret}$ 的数字中有多余的 $\text{guessDigit}$，因此将奶牛的数量加 $1$；

3. 将 $\text{secretDigit}$ 在哈希表中的出现次数加 $1$，将 $\text{guessDigit}$ 在哈希表中的出现次数减 $1$。

遍历结束之后，即可知道奶牛的数量。

代码

class Solution {
    public String getHint(String secret, String guess) {
        int bulls = 0, cows = 0;
        int[] digits = new int[10];
        int length = secret.length();
        for (int i = 0; i < length; i++) {
            int secretDigit = secret.charAt(i) - '0';
            int guessDigit = guess.charAt(i) - '0';
            if (secretDigit == guessDigit) {
                bulls++;
            } else {
                if (digits[secretDigit] < 0) {
                    cows++;
                }
                if (digits[guessDigit] > 0) {
                    cows++;
                }
                digits[secretDigit]++;
                digits[guessDigit]--;
            }
        }
        return String.format("%dA%dB", bulls, cows);
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是字符串 $\text{secret}$ 和 $\text{guess}$ 的长度。需要遍历字符串 $\text{secret}$ 和 $\text{guess}$ 各一次统计公牛数量和奶牛数量。

• 空间复杂度：$O(C)$，其中 $C$ 是数字的个数，$C=10$。需要使用一个哈希表记录每个数字的出现次数。