【leetcode】401 Binary Watch

二进制表,给出可能的时间


描述

有一块二进制表,4 个灯代表小时( 0 - 11 ), 6 个灯代表分钟( 00 - 59 )现在告诉你有 n 个灯亮了,你需要给出所有的可能时间。

样例

1
2
输入: 1
输出: ["1:00", "2:00", "4:00", "8:00", "0:01", "0:02", "0:04", "0:08", "0:16", "0:32"]

思路

😅 想直接枚举解决,一个一个列出来,一个灯的话 0 , 1 , 2 , … , 两个灯的话,01, 02 , 03 , … ? 这样列举十个灯实在太烦人了。于是换了一种 a + b 的方法,a代表小时灯亮的个数, b 代表分钟灯亮的个数,分别计算。开始想枚举 a = 0 , a = 1 ,后来发现这可以转化为统计一个数的二进制数 1 的个数, 豁然开朗。

测试数据可以用 vector 保存多个,一次性跑完所有测试集

代码

countBinary() 为统计二进制数中 1 的个数
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
class Solution {
public:
int countBinary(int num) {
int sum = 0;
while (num) {
num &= (num - 1);
sum++;
}
return sum;
}

vector<string> readBinaryWatch(int num) {
vector<string> result;
map<int, int> hour_map;
map<int, int> minute_map;
for (int i = 0; i < 12; i++) {
hour_map[i]=countBinary(i);
}
for (int i = 0; i < 60; i++) {
minute_map[i] = countBinary(i);
}
int len = num > 3 ? 3 : num;
for (int a = 0; a <= len; a++) {
vector<string> hour;
vector<string> minute;
int b = num - a;
for (auto p : hour_map) {
if (p.second == a) {
hour.push_back(to_string(p.first));
}
}

for (auto p : minute_map) {
if (p.second == b) {
if (p.first < 10) {
minute.push_back("0" + to_string(p.first));
}
else {
minute.push_back(to_string(p.first));
}
}
}

for (auto ehour : hour) {
for (auto eminute : minute) {
result.push_back(ehour + ":"+eminute);
}
}
}
return result;
}
};

参考

原题链接