python 珠玑妙算

珠玑妙算游戏（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次。

class Solution:
    """
    解体思路: 两次fo循环
    1. 第一次for循环过滤出猜中的个数
    2. 第二次for循环过滤出伪猜中的个数
    分别用s_list和g_list 来标记那些是猜中的和伪猜中的，不能重复使用
    """
    def masterMind(self, solution: str, guess: str) -> List[int]:
        s_list = [0, 0, 0, 0]
        g_list = [0, 0, 0, 0]

        for i in range(4):
            if solution[i] == guess[i]:
                s_list[i] = 1
                g_list[i] = 1

        for s_index, s_val in enumerate(solution):
            for g_index, g_val in enumerate(guess):
                if s_list[s_index]:
                    break
                if g_list[g_index]:
                    continue
                if s_val == g_val:
                    s_list[s_index] = -1
                    g_list[g_index] = -1
        return [s_list.count(1), s_list.count(-1)]