【leetcode】珠玑妙算（python实现）

忘是亡心i 2022-10-31 13:51 144阅读 0赞

题目：

珠玑妙算游戏（the game of master mind）的玩法如下。

计算机有4个槽，每个槽放一个球，颜色可能是红色（R）、黄色（Y）、绿色（G）或蓝色（B）。例如，计算机可能有RGGB 4种（槽1为红色，槽2、3为绿色，槽4为蓝色）。作为用户，你试图猜出颜色组合。打个比方，你可能会猜YRGB。要是猜对某个槽的颜色，则算一次“猜中”；要是只猜对颜色但槽位猜错了，则算一次“伪猜中”。注意，“猜中”不能算入“伪猜中”。

给定一种颜色组合solution和一个猜测guess，编写一个方法，返回猜中和伪猜中的次数answer，其中answer\[0\]为猜中的次数，answer\[1\]为伪猜中的次数。

示例：

输入： solution="RGBY",guess="GGRR"  
输出： \[1,1\]  
解释： 猜中1次，伪猜中1次。

来源：力扣（LeetCode）  
链接：https://leetcode-cn.com/problems/master-mind-lcci

思路：  
相同->猜中，从字符串中删除（直接在这一步就把猜中从伪猜中的数量中减去了）；

不同->保留（作为错误字符串进行后续处理）；

统计错误字符串中每个字符的数量，与正确字符串取交集即为伪猜中数量，然后删除对应字符。

代码：

class Solution:
        def masterMind(self, solution: str, guess: str) -> List[int]:
            j=0
            answer=[0,0]
            for _ in solution:
                if _==guess[j]:
                    answer[0]+=1
                    guess=guess.replace(_,"",1)
                    solution=solution.replace(_,"",1) 
                else:
                    j+=1 
            for _ in guess:
                if _ != "" :
                    count1 = guess.count(_)
                    count2 = solution.count(_)
                    if count1 > 1:
                        guess=list(filter(lambda x : x != _ , guess))   
                    answer[1]+=min(count2,count1)
            return answer

结果：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM3OTEwODQ4_size_16_color_FFFFFF_t_70][]

总结：代码长度较长 表现也一般 较为基础 综合运用了filter函数 （前为判断函数，后为需要迭代的内容）、count函数、replace函数，可参考并熟悉基础函数。

比较喜欢的他人代码：（非常简洁，思想基本与本人的相同，最终的执行用时和内存消耗经检测也与本人的一致，但实现方式简洁很多，值得借鉴）

class Solution:
        def masterMind(self, solution: str, guess: str) -> List[int]: 
             a = sum(i == j for i, j in zip(solution, guess))
             b = sum((collections.Counter(solution) & collections.Counter(guess)).values())
             return [a, b - a]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM3OTEwODQ4_size_16_color_FFFFFF_t_70]: /images/20221024/b40c9cb005334eea9ea2226a40cd55ab.png