题目在这:https://leetcode-cn.com/problems/bulls-and-cows/
题目分析:
这个题目描述的不是很好理解,我用一个例子简单的说明一下题目。
输入: secret = “1807”, guess = “7810”
输出: “1A3B”
对比两个字符串,从0开始遍历
索引0: 1 和 7 ,
索引1: 8 和 8 ,
索引2: 0 和 1 ,
索引3: 7 和 0.
显然 只有在索引1的位置上 两个字符串的字符相等,所以答案里 字母A前面那个数就是1。如果有两个这样的情况,相同位置上的字符相等,则A前面那个数就是2.。。(题目中叫做公牛)
而 索引0上的第一个串和索引2 上的第二个串都是1.所以此时B后面的那个数就是1,如果再有位置不同,但字符相同的情况,B后面的那个数就再加1。。。。(题目中叫做母牛)
一个数字成为了公牛,则他就不能是母牛。 所以这个例子里,索引1 位置相同,字符相同,公牛数为1.
107三个字符都是 有相同的字符,但位置不同,所以母牛数为3.最后返回 1A3B。
思路分析;
我们将第二个字符串构造哈希表 hs, key存字符值,val存该字符出现的次数。
首先遍历第一个字符串,如果在相同位置上第二个字符串也有相同的字符,则公牛数+1,并且在哈希表中去掉这一个字符。并且记录当前下标位置,(即当前位置已经是公牛了,后续不再考虑该位置元素。)
再次遍历第一个字符串,如果当前元素在哈希表中,并且相应val大于0,则母牛数+1,相应的字符个数减1.
完整代码
class Solution:
def getHint(self, secret: str, guess: str) -> str:
bull = 0
cow = 0
from collections import Counter
hs_gu = Counter(guess)
temp = [] # 记录当前为公牛的数字下标
for i in range(len(secret)):
if secret[i] == guess[i]:
bull +=1
hs_gu[secret[i]] -=1
temp.append(i)
for j in range(len(secret)):
if secret[j] in hs_gu and j not in temp:
if hs_gu[secret[j]] > 0:
cow +=1
hs_gu[secret[j]] -=1
return str(bull) +'A' + str(cow) + 'B'