LeetCode 299 Bulls and Cows

题意

你正和你的朋友一起玩下面的公牛和母牛游戏:你写下一个数字然后让你的朋友猜猜这个数字是多少. 每当你的朋友猜测时, 你提供一个提示, 表明所述猜测中有多少位数与你的密码完全匹配,包括数字和位置(称为”公牛”)以及有多少位数与密码匹配但位于错误的位置(称为”奶牛”)。
编写一个函数, 根据秘密数字和朋友的猜测返回提示, 用于 A 表示公牛, B 表示奶牛.

例 1:

1
2
3
4
5
输入: secret = "1807", guess = "7810"

输出: "1A3B"

说明: 1 个公牛和 3 个奶牛. 公牛是 8, 奶牛是 0, 1 和 7.

例 2:

1
2
3
4
5
输入: 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) 的空间复杂度.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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.

  • 本文作者: 赵俊
  • 本文链接: http://www.zhaojun.im/leetcode-299/
  • 版权声明: 本博客所有文章除特别声明外,均采用 BY-NC-SA 许可协议。转载请注明出处!