猜数字游戏

2019-03-25  本文已影响0人  Haward_

你正在和你的朋友玩 [猜数字(Bulls and Cows)]游戏:你写下一个数字让你的朋友猜。每次他猜测后,你给他一个提示,告诉他有多少位数字和确切位置都猜对了(称为“Bulls”, 公牛),有多少位数字猜对了但是位置不对(称为“Cows”, 奶牛)。你的朋友将会根据提示继续猜,直到猜出秘密数字。

请写出一个根据秘密数字和朋友的猜测数返回提示的函数,用 A 表示公牛,用 B 表示奶牛。

请注意秘密数字和朋友的猜测数都可能含有重复数字。

示例 1:

输入: secret = "1807", guess = "7810"

输出: "1A3B"

解释: 1 公牛和 3 奶牛。公牛是 8,奶牛是 0, 1 和 7。
示例 2:

输入: secret = "1123", guess = "0111"

输出: "1A1B"

解释: 朋友猜测数中的第一个 1 是公牛,第二个或第三个 1 可被视为奶牛。
说明: 你可以假设秘密数字和朋友的猜测数都只包含数字,并且它们的长度永远相等

解:hash表,时间复杂度O(n)

class Solution:
    def getHint(self, secret, guest):
        if len(secret)!=len(guest):
            return "0A0B"
        d_res = {'A':0,'B':0}
        d = {}
        for i in range(len(secret)):
            if secret[i] == guest[i]:
                d_res['A'] += 1
            else:
                d[secret[i]] = d[secret[i]]+1 if d.get(secret[i])!=None else 1
                
        for i in range(len(secret)):
            if secret[i] != guest[i] and d.get(guest[i])!=None and d.get(guest[i])>0:
                d_res['B']+=1
                d[guest[i]]-=1
                
        return str(d_res['A'])+'A'+str(d_res['B'])+'B'
                
上一篇 下一篇

猜你喜欢

热点阅读