You are playing the following Bulls and Cows game with your friend: You write a 4-digit secret number and ask your friend to guess it. Each time your friend guesses a number, you give a hint. The hint tells your friend how many digits are in the correct positions (called “bulls”) and how many digits are in the wrong positions (called “cows”). Your friend will use those hints to find out the secret number.
For example:
Secret number: “1807”
Friend’s guess: “7810”
Hint: 1 bull and 3 cows. (The bull is 8, the cows are 0, 1 and 7.)
Write a function to return a hint according to the secret number and friend’s guess, use A to indicate the bulls and B to indicate the cows. In the above example, your function should return "1A3B"
.
Please note that both secret number and friend’s guess may contain duplicate digits, for example:
Secret number: “1123”
Friend’s guess: “0111”
In this case, the 1st 1 in friend’s guess is a bull, the 2nd or 3rd 1 is a cow, and your function should return "1A1B"
.
You may assume that the secret number and your friend’s guess only contain digits, and their lengths are always equal.
解题思路
- 扫描secret字符串,记录各个字符的个数。
- 扫描两遍guess字符串:
①第一遍记录字符正确且位置也正确的字符。
②第二遍记录字符正确但位置不正确的字符。
实现代码
C++:
// Runtime: 20ms
class Solution {
public:
string getHint(string secret, string guess) {
unordered_map<char, int> mymap;
vector<bool> tag(secret.size(), false);
for (int i = 0; i < secret.size(); i++) {
mymap[secret[i]]++;
}
int cntA = 0;
int cntB = 0;
for (int i = 0; i < guess.size(); i++) {
if (secret[i] == guess[i]) {
++cntA;
tag[i] = true;
mymap[guess[i]]--;
}
}
for (int i = 0; i < guess.size(); i++) {
if (!tag[i] && mymap[guess[i]] > 0) {
cntB++;
mymap[guess[i]]--;
}
}
return to_string(cntA) + "A" + to_string(cntB) + "B";
}
private:
string to_string(int n) {
if (n == 0)
{
return "0";
}
string res = "";
while (n)
{
res += n % 10 + '0';
n /= 10;
}
reverse(res.begin(), res.end());
return res;
}
};
Java:
// Runtime: 18ms
public class Solution {
public String getHint(String secret, String guess) {
Map<Character, Integer> map = new HashMap<Character, Integer>();
boolean tag[] = new boolean[secret.length()];
for (int i = 0; i < secret.length(); i++) {
if (map.containsKey(secret.charAt(i))) {
map.replace(secret.charAt(i), map.get(secret.charAt(i)) + 1);
}
else {
map.put(secret.charAt(i), 1);
}
}
int a = 0, b = 0;
for (int i = 0; i < guess.length(); i++) {
if(secret.charAt(i) == guess.charAt(i)) {
++a;
map.replace(guess.charAt(i), map.get(guess.charAt(i)) - 1);
tag[i] = true;
}
}
for (int i = 0; i < guess.length(); i++) {
if (!tag[i] && map.containsKey(guess.charAt(i)) && map.get(guess.charAt(i)) > 0) {
++b;
map.replace(guess.charAt(i), map.get(guess.charAt(i)) - 1);
}
}
return a + "A" + b + "B";
}
}