[Python] BOJ 2564번. 경비원

작성일 | In |

2564번. 경비원

문제 링크

풀이 코드

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
# 2564번. 경비원


import sys
input = sys.stdin.readline

r, c = map(int, input().split())


def getDistanceFromZero(dir, d):
    if dir == 1:  # 북
        return d
    elif dir == 2:  # 남
        return r+c+r-d
    elif dir == 3:  # 서
        return r+c+r+c-d
    else:  # 동
        return r+d


def getDistance(loc, dong):
    total = r*2+c*2  # 둘레
    # 두가지 방면으로 구하기
    return min(abs(dong-loc), total-abs(dong-loc))


n = int(input())
nList = []
for i in range(n):
    dir, d = map(int, input().split())  # 방향, 거리
    nList.append(getDistanceFromZero(dir, d))  # 왼쪽 위 모서리로부터 거리

dir, d = map(int, input().split())
dong = getDistanceFromZero(dir, d)  # 동근이

ans = 0
for i in range(n):
    ans += getDistance(nList[i], dong)  # 최단경로 구해서 더해주기

print(ans)

비고