299. 猜数字游戏

你在和朋友一起玩 猜数字(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"

 

posted @ 2021-11-08 10:27  散装英语king  阅读(417)  评论(0编辑  收藏  举报