[알고리즘]BOJ_2564_경비원(파이썬)

#문제 링크

https://www.acmicpc.net/problem/2564

2564번: 경비원

 

#문제

https://www.acmicpc.net/problem/2564
경비원 

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
1 초	128 MB	3790	2010	1592	53.209%
문제
동근이는 무인 경비 회사 경비원으로 항상 대기하고 있다가 호출이 들어오면
경비차를 몰고 그 곳으로 달려가야 한다.
동근이가 담당하고 있는 곳은 직사각형 모양의 블록으로 블록 중간을 가로질러 차가 통과할만한 길이 없다.
이 블록 경계에 무인 경비를 의뢰한 상점들이 있다.

예를 들어 가로의 길이가 10, 세로의 길이가 5인 블록의 경계에 무인 경비를 의뢰한 3개의 상점이 있다고 하자.
 <그림 1>과 같이 이들은 1, 2, 3으로 표시되어 있고, 동근이는 X로 표시한 위치에 있다.



1번 상점에서 호출이 들어 왔을 때 동근이가 블록을 시계방향으로 돌아 이동하면 이동 거리가 12가 된다.
 반면 반시계방향으로 돌아 이동하면 이동 거리는 18이 된다.
 따라서 동근이가 1번 상점으로 가는 최단 거리는 12가 된다.
 마찬가지로 동근이의 위치에서 2번 상점까지의 최단 거리는 6, 3번 상점까지의 최단 거리는 5가 된다.

블록의 크기와 상점의 개수 및 위치 그리고 동근이의 위치가 주어질 때
동근이의 위치와 각 상점 사이의 최단 거리의 합을 구하는 프로그램을 작성하시오.

입력
첫째 줄에 블록의 가로의 길이와 세로의 길이가 차례로 주어진다.
둘째 줄에 상점의 개수가 주어진다. 블록의 가로의 길이와 세로의 길이,
상점의 개수는 모두 100이하의 자연수이다. 이어 한 줄에 하나씩 상점의 위치가 주어진다.
상점의 위치는 두 개의 자연수로 표시된다.
첫째 수는 상점이 위치한 방향을 나타내는데, 1은 블록의 북쪽, 2는 블록의 남쪽, 3은 블록의 서쪽, 4는 블록의 동쪽에
상점이 있음을 의미한다.
둘째 수는 상점이 블록의 북쪽 또는 남쪽에 위치한 경우 블록의 왼쪽 경계로부터의 거리를 나타내고,
상점이 블록의 동쪽 또는 서쪽에 위치한 경우 블록의 위쪽 경계로부터의 거리를 나타낸다.
마지막 줄에는 동근이의 위치가 상점의 위치와 같은 방식으로 주어진다.
상점의 위치나 동근이의 위치는 블록의 꼭짓점이 될 수 없다.

출력
첫째 줄에 동근이의 위치와 각 상점 사이의 최단 거리의 합을 출력한다.

예제 입력 1
10 5
3
1 4
3 2
2 8
2 3
예제 출력 1
23

 

#답안

import sys
input=sys.stdin.readline

#북쪽 왼쪽 모서리를 0으로 잡고 펼쳤을 때 0으로부터 얼마나 떨어졌는 지 위치를 계산해주는 함수
def cal_dist(loc, distance) :
    if loc==1 : #북쪽
        return distance
    elif loc==4 : #동쪽
        return w+distance
    elif loc==2 : #남쪽
        return w+h+w-distance
    elif loc==3 : #서쪽
        return w+h+w+h-distance


#블록의 가로길이 세로길이 입력받기
w, h = map(int,input().split())

#상점의 개수 입력받기
num=int(input())

#상점의 위치를 저장할 리스트 만들어 놓기
locations=[0]*(num+1)

dist=[]
#각 상점의 위치와 동근이의 위치 입력받기
for i in range(num+1) :
    loc, distance=map(int, input().split())
    dist.append(cal_dist(loc, distance))

#동근이의 0에서 부터 떨어진 위치를 저장
dong_dist=dist[-1]


answer=0
for i in range(num):
    #0에서 떨어진 각 가게의 거리와, 0에서 떨어진 동근의 위치 차이의 절댓값을 구해준다.
    cal_distance=abs(dist[i]-dong_dist)

    #전체 길이를 구해준다.
    total=2*(w+h)

    #더 작은 값을 answer에 누적시킨다.
    answer+=min(cal_distance,total-cal_distance)

print(answer)

👆풀이

1. 동서남북 박스를 쫙 펼쳤다고 생각하고 각 좌표를 다시 구하자.

2. 각 좌표와 동근의 좌표 차의 절대값을 거리a라고 하면 (전체 길이-a)는 a와의 반대방향의 거리인 b가 나올 것이다.

3. a와 b를 비교해서 min값을 answer에 누적시킨다.