299. 猜数字游戏(python)

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

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

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

输入: secret = "1807", guess = "7810"
输出: "1A3B"
解释: 1 公牛和 3 奶牛。公牛是 8,奶牛是 0, 1 和 7

输入: secret = "1123", guess = "0111"
输出: "1A1B"
解释: 朋友猜测数中的第一个 1 是公牛,第二个或第三个 1 可被视为奶牛。

AC

思路

— 同时遍历secret、guess,对应位置相等则为公牛
— 找奶牛时去掉公牛,否则会造成干扰
— 建两个dict, 分别存储secret、guess中出现的数字及其个数。
— 遍历secret中每个key值,遍历是否出现在guess,是的话就是奶牛,取这个key值在两dict中的最小值

class Solution:
    def getHint(self, secret: str, guess: str) -> str:
        bull = 0
        cow = 0
        dic_secret = {}
        dic_guess = {}
        for i in range(len(secret)):
            #找公牛
            if secret[i] == guess[i]:
                bull += 1
            else:
                #去除公牛,分别构建dict存储 secret、guess中出现的数字及其个数
                if secret[i] not in dic_secret:
                    dic_secret[secret[i]] = 1
                else:
                    dic_secret[secret[i]] += 1
                if guess[i] not in dic_guess:
                    dic_guess[guess[i]] = 1
                else:
                    dic_guess[guess[i]] += 1
        # 如果secret中的key出现在guess中,取两dict中相应key的较小value为奶牛数
        for i in dic_secret:
            if i in dic_guess:
                cow += min(dic_secret[i], dic_guess[i])
        return str(bull)+'A'+str(cow)+'B'

复杂度O(n^2)

评论优解

思路

— 都是数字,因此可以直接用list存,list的索引即key, 值为出现的次数value
— list索引:ord(key) - ord(‘0’) = 1 - 0 / 2- 0 / 3- 0 /…
— 计算奶牛次数时挨个遍历两list,取相应位置较小的值相加,没出现的都是0

class Solution:
    def getHint(self, secret: str, guess: str) -> str:
        bull = 0
        cow = 0
        dic_secret = [0] * 10
        dic_guess = [0] * 10
        for i in range(len(secret)):
            if secret[i] == guess[i]:
                bull += 1
            else:
                dic_secret[ord(secret[i]) - ord('0')] += 1
                dic_guess[ord(guess[i]) - ord('0')] += 1
        for i in range(len(dic_secret)):
            cow += min(dic_secret[i], dic_guess[i])
        return str(bull)+'A'+str(cow)+'B'

复杂度O(n)

展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 深蓝海洋 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读