珠玑妙算游戏(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种字符
package Solutionmst1615;
import java.util.Arrays;
class Solution {
public int[] masterMind(String solution, String guess) {
int guessRcount = 0;
int guessYcount = 0;
int guessGcount = 0;
int guessBcount = 0;
int solutionRcount = 0;
int solutionYcount = 0;
int solutionGcount = 0;
int solutionBcount = 0;
for (int i = 0; i < guess.length(); i++) {
if (guess.charAt(i) == 'R') {
guessRcount++;
}
if (guess.charAt(i) == 'Y') {
guessYcount++;
}
if (guess.charAt(i) == 'G') {
guessGcount++;
}
if (guess.charAt(i) == 'B') {
guessBcount++;
}
}
for (int i = 0; i < solution.length(); i++) {
if (solution.charAt(i) == 'R') {
solutionRcount++;
}
if (solution.charAt(i) == 'Y') {
solutionYcount++;
}
if (solution.charAt(i) == 'G') {
solutionGcount++;
}
if (solution.charAt(i) == 'B') {
solutionBcount++;
}
}
int[] out = new int[2];
for (int i = 0; i < guess.length(); i++) {
for (int j = 0; j < solution.length(); j++) {
if (guess.charAt(i) == solution.charAt(j) && i == j) {
if (solution.charAt(i) == 'R') {
solutionRcount--;
guessRcount--;
}
if (solution.charAt(i) == 'Y') {
solutionYcount--;
guessYcount--;
}
if (solution.charAt(i) == 'G') {
solutionGcount--;
guessGcount--;
}
if (solution.charAt(i) == 'B') {
solutionBcount--;
guessBcount--;
}
out[0]++;
}
}
}
if (guessRcount > 0 && solutionRcount > 0) {
out[1] = out[1] + Math.min(guessRcount, solutionRcount);
}
if (guessYcount > 0 && solutionYcount > 0) {
out[1] = out[1] + Math.min(guessYcount, solutionYcount);
}
if (guessGcount > 0 && solutionGcount > 0) {
out[1] = out[1] + Math.min(guessGcount, solutionGcount);
}
if (guessBcount > 0 && solutionBcount > 0) {
out[1] = out[1] + Math.min(guessBcount, solutionBcount);
}
return out;
}
public static void main(String[] args) {
Solution sol = new Solution();
String solution = "YBBY", guess = "GYYB";
System.out.println(Arrays.toString(sol.masterMind(solution, guess)));
}
}