【Leetcode】299. 猜数字游戏

1,087 阅读2分钟

题目描述

你正在和你的朋友玩 猜数字(Bulls and Cows)游戏:你写下一个数字让你的朋友猜。每次他猜测后,你给他一个提示,告诉他有多少位数字和确切位置都猜对了(称为“Bulls”, 公牛),有多少位数字猜对了但是位置不对(称为“Cows”, 奶牛)。你的朋友将会根据提示继续猜,直到猜出秘密数字。

请写出一个根据秘密数字和朋友的猜测数返回提示的函数,用 A 表示公牛,用 B 表示奶牛。

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

示例 1:

输入: secret = "1807", guess = "7810"

输出: "1A3B"

解释: 1 公牛和 3 奶牛。公牛是 8,奶牛是 0, 1 和 7。 示例 2:

输入: secret = "1123", guess = "0111"

输出: "1A1B"

解释: 朋友猜测数中的第一个 1 是公牛,第二个或第三个 1 可被视为奶牛。 说明: 你可以假设秘密数字和朋友的猜测数都只包含数字,并且它们的长度永远相等。

题解

用两个数组(大小为为10)记录0-9数字的个数。遍历一次字符串,比较相同位置数字是否相等,相等则公牛数加1,不相等则将该位置数字存入到数组中。最后在两个数组中相同位置较小数相加,即为奶牛数。

class Solution {
    public String getHint(String secret, String guess) {
        int aCount = 0;
        int bCount = 0;
        int[] sDatas = new int[10];
        int[] gDatas = new int[10];
        for (int i = 0; i < secret.length(); i++) {
            int sPos = secret.charAt(i) - '0';
            int gPos = guess.charAt(i) - '0';
            if (secret.charAt(i) == guess.charAt(i)) {
                aCount++;
            } else {
                sDatas[sPos] = sDatas[sPos] + 1;
                gDatas[gPos] = gDatas[gPos] + 1;
            }
        }
        for (int i = 0; i < sDatas.length; i++) {
            int min = Math.min(sDatas[i], gDatas[i]);
            bCount += min;
        }
        StringBuilder res = new StringBuilder();
        res.append(aCount).append("A");
        res.append(bCount).append("B");
        return res.toString();
    }
}