-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathByteDance_1.py
85 lines (75 loc) · 2.24 KB
/
ByteDance_1.py
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
# -*- coding:utf-8 -*-
""" 雀魂启动 (字节跳动)"""
import copy
num = map(int, input().split())
num_map = dict([(i, 0) for i in range(1, 10)])
result = []
for i in num:
num_map[i] += 1
# 假设添加1-9
for i in range(1, 10):
flag = False
if num_map[i] == 4:
continue
num_map[i] += 1
# 选择雀头
for j in range(1, 10):
tmp = copy.copy(num_map)
if tmp[j] < 2:
continue
tmp[j] -= 2
# 判断能否胡牌
for k in range(1, 10):
print(tmp)
if tmp[k] == 0:
flag = True
continue
elif tmp[k] <= 2:
k_num = tmp[k]
val_1 = k + 1
val_2 = k + 2
if val_1 in tmp and val_2 in tmp and tmp[val_1] >= k_num and tmp[val_2] >= k_num:
flag = True
tmp[val_1] -= k_num
tmp[val_2] -= k_num
else:
flag = False
break
elif tmp[k] == 3:
k_num = tmp[k]
val_1 = k + 1
val_2 = k + 2
flag = True
if val_1 in tmp and val_2 in tmp and tmp[val_1] >= k_num and tmp[val_2] >= k_num:
tmp[val_1] -= k_num
tmp[val_2] -= k_num
else:
tmp[k] -= 3
else:
k_num = tmp[k]
val_1 = k + 1
val_2 = k + 2
if val_1 in tmp and val_2 in tmp and tmp[val_1] >= k_num and tmp[val_2] >= k_num:
flag = True
tmp[val_1] -= k_num
tmp[val_2] -= k_num
elif val_1 in tmp and val_2 in tmp and tmp[val_1] >= 1 and tmp[val_2] >= 1:
flag = True
tmp[val_1] -= 1
tmp[val_2] -= 1
tmp[k] -= 3
else:
flag = False
break
tmp = copy.copy(num_map)
if flag:
break
num_map[i] -= 1
# 收录结果
if flag:
result.append(i)
if len(result) == 0:
print(0)
else:
result = map(str, result)
print(" ".join(result))