你正和你的朋友一起玩下面的公牛和母牛游戏:你写下一个数字然后让你的朋友猜猜这个数字是多少. 每当你的朋友猜测时, 你提供一个提示, 表明所述猜测中有多少位数与你的密码完全匹配,包括数字和位置(称为”公牛”)以及有多少位数与密码匹配但位于错误的位置(称为”奶牛”)。 编写一个函数, 根据秘密数字和朋友的猜测返回提示, 用于 A 表示公牛, B 表示奶牛.
例 1:
输入: secret = "1807", guess = "7810"
输出: "1A3B"
说明: 1 个公牛和 3 个奶牛. 公牛是 8, 奶牛是 0, 1 和 7.
例 2:
输入: secret = "1123", guess = "0111"
输出: "1A1B"
说明: The 1st 1 in friend's guess is a bull, the 2nd or 3rd 1 is a cow.
刚开始我的想法是依次获取公牛和奶牛的数量, 但奶牛的判断需要 O(n^2) 的时间复杂度, 后面想到, 用所有匹配的数量 - 公牛的数量就是奶牛的数量, 只需要 O(n) 的时间复杂度和 O(1) 的空间复杂度.
public String getHint(String secret, String guess) {
int[] table = new int[10];
int total = 0;
int bulls = 0;
for (char c :secret.toCharArray()) {
table[c - '0']++;
}
for (int i = 0; i < guess.length(); i++) {
if (secret.charAt(i) == guess.charAt(i)) {
bulls++;
}
if (table[guess.charAt(i) - '0']-- > 0) {
total++;
}
}
return bulls + "A" + (total - bulls) + "B";
}
Runtime: 1 ms, faster than 100.00% of Java online submissions for Bulls and Cows.