面试题 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;
}