-
Notifications
You must be signed in to change notification settings - Fork 0
/
양궁대회.py
48 lines (41 loc) · 1.61 KB
/
양궁대회.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
def dfs(shot, apeach_board, idx, apeach_score, lion_score, lion_board):
if shot == 0:
global max_val, answer
if lion_score - apeach_score > max_val:
max_val = lion_score - apeach_score
answer = lion_board[:]
elif lion_score - apeach_score == max_val and max_val != 0:
for i in range(10, -1, -1):
if lion_board[i] > answer[i]:
answer = lion_board[:]
break
elif lion_board[i] < answer[i]:
break
return
for i in range(idx, 11):
if shot > apeach_board[i]:
lion_board[i] = apeach_board[i] + 1
if apeach_board[i] > 0:
dfs(shot - lion_board[i], apeach_board, i + 1, apeach_score - (10 - i), lion_score + (10 - i), lion_board)
else:
dfs(shot - lion_board[i], apeach_board, i + 1, apeach_score, lion_score + (10 - i), lion_board)
lion_board[i] = 0
else:
if i == 10:
lion_board[i] = shot
dfs(0, apeach_board, i + 1, apeach_score, lion_score, lion_board)
lion_board[i] = 0
else:
dfs(shot, apeach_board, i + 1, apeach_score, lion_score, lion_board)
def solution(n, info):
global max_val, answer
apeach_score = 0
answer = [0 for _ in range(11)]
max_val = 0
for i in range(11):
if info[i] > 0:
apeach_score += 10 - i
dfs(n, info, 0, apeach_score, 0, [0 for _ in range(11)])
if answer == [0 for _ in range(11)]:
answer = [-1]
return answer