LeetCode 299. Bulls and Cows
考点 | 难度 |
---|---|
Hash Map | Medium |
题目
You are playing the Bulls and Cows game with your friend.
You write down a secret number and ask your friend to guess what the number is. When your friend makes a guess, you provide a hint with the following info:
The number of “bulls”, which are digits in the guess that are in the correct position.
The number of “cows”, which are digits in the guess that are in your secret number but are located in the wrong position. Specifically, the non-bull digits in the guess that could be rearranged such that they become bulls.
Given the secret number secret
and your friend’s guess guess
, return the hint for your friend’s guess.
The hint should be formatted as "xAyB"
, where x
is the number of bulls and y
is the number of cows. Note that both secret
and guess
may contain duplicate digits.
思路
先算B,再算B + C。对于每一个数字,secret和guess里出现次数更小的数就是那个数字的B + C,最后减掉B得到C
答案
class Solution:
def getHint(self, secret, guess):
B = sum([x == y for x,y in zip(secret, guess)])
count_sec = Counter(secret)
count_gue = Counter(guess)
B_C = sum([min(count_sec[elem], count_gue[elem]) for elem in count_sec])
return str(B) + "A" + str(B_C-B) + "B"