面试题 16.15. 珠玑妙算

题目

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

提示：

len(solution) = len(guess) = 4
solution和guess仅包含”R”,”G”,”B”,”Y”这4种字符

题解

public int[] masterMind(String solution, String guess) {
    // 按照字符计数、猜中及伪猜中
    int[] counter = new int[26], result = new int[2];
    // 计数猜测颜色
    for (int i = 0; i < 4; i++) {
        char g = guess.charAt(i);
        // 计数颜色出现次数
        counter[g - 'A']++;
        if (g == solution.charAt(i)) {
            // 猜中计数
            result[0]++;
            // 颜色计数减1
            counter[g - 'A']--;
        }
    }

    // 根据猜测计数判断伪猜中
    for (int i = 0; i < 4; i++) {
        char s = solution.charAt(i);
        if (guess.charAt(i) != s && counter[s - 'A'] > 0) {
            // 伪猜中计数
            result[1]++;
            // 颜色计数减1
            counter[s - 'A']--;
        }
    }

    return result;
}