你在和朋友一起玩 猜数字(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 可以成为公牛数字。
示例 3:
输入:secret = "1", guess = "0"
输出:"0A0B"
示例 4:
输入:secret = "1", guess = "1"
输出:"1A0B"
提示:
1 <= secret.length, guess.length <= 1000
secret.length == guess.length
secret 和 guess 仅由数字组成
思路1:
字典+拉链法
将数字和和出现的位置记录在字典中,在遍历guess计算A和B的值
1 class Solution: 2 def getHint(self, secret: str, guess: str) -> str: 3 maps = {} 4 length = len(secret) 5 bull_mask = [False] * length 6 A = 0 7 B = 0 8 for i in range(length): 9 key = secret[i] 10 if key in maps: 11 maps[key].append(i) 12 else: 13 maps[key] = [i] 14 15 # bulls 16 for i in range(length): 17 guess_ch = guess[i] 18 if guess_ch in maps and i in maps[guess_ch]: 19 A += 1 20 maps[guess_ch].remove(i) 21 bull_mask[i] = True 22 23 # cows 24 for i in range(length): 25 guess_ch = guess[i] 26 if not bull_mask[i] and guess_ch in maps and maps[guess_ch]: 27 B += 1 28 maps[guess_ch].pop() 29 return format(f"{A}A{B}B")
思路2:
考虑到数字一共只有10种出现情况,可以将字典简化为数组
某数字对应奶牛的数量是 排除公牛后,min(guess该数字出现次数,secret中该数字出现次数)
1 class Solution2: 2 def getHint(self, secret: str, guess: str) -> str: 3 A = 0 4 B = 0 5 cnt_secret = [0] * 10 6 cnt_guess = [0] * 10 7 8 for i in range(len(secret)): 9 guess_ch = guess[i] 10 secret_ch = secret[i] 11 # bulls 12 if guess_ch == secret_ch: 13 A += 1 14 # cows 15 else: 16 cnt_secret[ord(secret_ch) - ord('0')] += 1 17 cnt_guess[ord(guess_ch) - ord('0')] += 1 18 for s, g in zip(cnt_secret, cnt_guess): 19 B += min(s, g) 20 return f"{A}A{B}B"