[Python] BOJ 19637번. IF문 좀 대신 써줘

작성일 | In |

19637번. IF문 좀 대신 써줘

문제 링크

풀이 코드

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
# 19637번. IF문 좀 대신 써줘


import sys
input = sys.stdin.readline
# 다시 풀어볼 문제...
n, m = map(int, input().split())

cost = []  # 전투력 관리
name = []  # 칭호만 관리
for i in range(n):
    a, b = input().rstrip().split()
    cost.append(int(b))
    name.append(a)
i = 0
while i < len(cost)-1:
    if cost[i] == cost[i+1]:
        cost.pop(i+1)
        name.pop(i+1)
    i += 1

t = []
for i in range(m):
    t.append([int(input()), i, 0])
t.sort()
for i in range(len(cost)):
    if t[0][0] <= cost[i]:
        t[0][2] = name[i]
        idx = i
        break

for i in range(m):
    if t[i][0] <= cost[idx]:
        t[i][2] = name[idx]
    else:
        idx += 1
        while True:
            if t[i][0] <= cost[idx]:
                t[i][2] = name[idx]
                break
            idx += 1
ans = [0 for i in range(m)]

for i in range(m):
    ans[t[i][1]] = t[i][2]
for i in range(m):
    print(ans[i])

비고