forked from haoel/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
bullsAndCows.cpp
90 lines (82 loc) · 3.34 KB
/
bullsAndCows.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
// Source : https://leetcode.com/problems/bulls-and-cows/
// Author : Calinescu Valentin, Hao Chen
// Date : 2015-11-04
/***************************************************************************************
*
* 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.
*
* Credits:
* Special thanks to @jeantimex for adding this problem and creating all test cases.
*
***************************************************************************************/
class Solution {
public:
string getHint(string secret, string guess) {
return getHint02(secret, guess);
return getHint01(secret, guess);
}
string getHint01(string secret, string guess) {
int appears_in_secret[10] = {0}, appears_in_guess[10] = {0}, bulls[10] = {0};
int total_bulls = 0, total_cows = 0;
for(int i = 0; i < secret.size(); i++)
appears_in_secret[secret[i] - '0']++;
for(int i = 0; i < guess.size(); i++)
{
if(secret[i] == guess[i])
bulls[secret[i] - '0']++;
appears_in_guess[guess[i] - '0']++;
}
for(int i = 0; i <= 9; i++)
{
total_bulls += bulls[i];
total_cows += min(appears_in_secret[i], appears_in_guess[i]) - bulls[i];
}
return to_string(total_bulls) + "A" + to_string(total_cows) + "B";
}
// Another implemntation - to save more space
string getHint02(string secret, string guess) {
const int digital_num = 10;
int secret_stat[digital_num]={0};
int bull = 0;
for(int i=0; i<secret.size(); i++) {
//both number & location are matched, count bull
if ( secret[i] == guess[i] ) {
bull++;
continue;
}
//otherwise, count the unmatched digits.
secret_stat[secret[i]-'0']++;
}
int cow = 0;
for(int i=0; i<guess.size(); i++) {
//deal with duplication - decrease the digits count if cow is found.
if ( secret[i] != guess[i] && secret_stat[guess[i]-'0']-- > 0 ) {
cow++;
}
}
return to_string(bull) + "A" + to_string(cow) + "B";
}
};