题目
你在和朋友一起玩 猜数字(Bulls and Cows)游戏,该游戏规则如下:
你写出一个秘密数字,并请朋友猜这个数字是多少。
朋友每猜测一次,你就会给他一个提示,告诉他的猜测数字中有多少位属于数字和确切位置都猜对了(称为“Bulls”, 公牛),有多少位属于数字猜对了但是位置不对(称为“Cows”, 奶牛)。
朋友根据提示继续猜,直到猜出秘密数字。
请写出一个根据秘密数字和朋友的猜测数返回提示的函数,返回字符串的格式为 xAyB ,x 和 y 都是数字,A 表示公牛,用 B 表示奶牛。
xA 表示有 x 位数字出现在秘密数字中,且位置都与秘密数字一致。
yB 表示有 y 位数字出现在秘密数字中,但位置与秘密数字不一致。
请注意秘密数字和朋友的猜测数都可能含有重复数字,每位数字只能统计一次。
示例
输入: secret = “1807”, guess = “7810”
输出: “1A3B”
解释: 1 公牛和 3 奶牛。公牛是 8,奶牛是 0, 1 和 7。
输入: secret = “1123”, guess = “0111”
输出: “1A1B”
解释: 朋友猜测数中的第一个 1 是公牛,第二个或第三个 1 可被视为奶牛。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/bulls-and-cows
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1
Python实现
class Solution:
def getHint(self, secret: str, guess: str) -> str:
sumA,sumB,n=0,0,len(guess)
for i in range(n):
if secret[i]==guess[i]:
sumA+=1
c1=collections.Counter(secret)
c2=collections.Counter(guess)
for i,v in c2.items():
sumB+=min(c2[i],c1[i])
sumB = str(sumB-sumA)
sumA = str(sumA)
return sumA + 'A' + sumB +'B'
Java实现1
class Solution {
public String getHint(String secret, String guess) {
int[] secretArr = new int[10];
int[] guessArr = new int[10];
// 获取公牛个数
int countA = 0;
for (int i = 0; i < secret.length(); i++) {
if (secret.charAt(i) == guess.charAt(i)) {
countA++;
}
// 获取每个数字个数
secretArr[secret.charAt(i) - '0']++;
guessArr[guess.charAt(i) - '0']++;
}
// 获取相同数字个数
int countAll = 0;
for (int i = 0; i < 10; i++) {
countAll += Math.min(secretArr[i], guessArr[i]);
}
// 获取母牛个数
int countB = countAll - countA;
return String.valueOf(countA) + "A" + String.valueOf(countB) + "B";
}
}
Java实现2
2个map来存。注意使用 getOrDefault() 方法
class Solution {
public String getHint(String secret, String guess) {
int bull = 0, cow = 0;
Map<Character, Integer> map1 = new HashMap<>();
Map<Character, Integer> map2 = new HashMap<>();
for (int i = 0; i < secret.length(); i++){
if (secret.charAt(i) == guess.charAt(i)) bull++; //公牛
map1.put(secret.charAt(i), map1.getOrDefault(secret.charAt(i), 0) + 1);
map2.put(guess.charAt(i), map2.getOrDefault(guess.charAt(i), 0) + 1);
}
for (Character key : map2.keySet()){
cow += Math.min(map2.getOrDefault(key, 0), map1.getOrDefault(key, 0));
}
cow = cow - bull;
return bull + "A" + cow + "B";
}
}