백준 : 9205(맥주 마시면서 걸어가기) [파이썬]

 

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

9205번: 맥주 마시면서 걸어가기

1.  문제 설명

문제

송도에 사는 상근이와 친구들은 송도에서 열리는 펜타포트 락 페스티벌에 가려고 한다. 올해는 맥주를 마시면서 걸어가기로 했다. 출발은 상근이네 집에서 하고, 맥주 한 박스를 들고 출발한다. 맥주 한 박스에는 맥주가 20개 들어있다. 목이 마르면 안되기 때문에 50미터에 한 병씩 마시려고 한다. 즉, 50미터를 가려면 그 직전에 맥주 한 병을 마셔야 한다.

상근이의 집에서 페스티벌이 열리는 곳은 매우 먼 거리이다. 따라서, 맥주를 더 구매해야 할 수도 있다. 미리 인터넷으로 조사를 해보니 다행히도 맥주를 파는 편의점이 있다. 편의점에 들렸을 때, 빈 병은 버리고 새 맥주 병을 살 수 있다. 하지만, 박스에 들어있는 맥주는 20병을 넘을 수 없다. 편의점을 나선 직후에도 50미터를 가기 전에 맥주 한 병을 마셔야 한다.

편의점, 상근이네 집, 펜타포트 락 페스티벌의 좌표가 주어진다. 상근이와 친구들이 행복하게 페스티벌에 도착할 수 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 t가 주어진다. (t ≤ 50)

각 테스트 케이스의 첫째 줄에는 맥주를 파는 편의점의 개수 n이 주어진다. (0 ≤ n ≤ 100).

다음 n+2개 줄에는 상근이네 집, 편의점, 펜타포트 락 페스티벌 좌표가 주어진다. 각 좌표는 두 정수 x와 y로 이루어져 있다. (두 값 모두 미터, -32768 ≤ x, y ≤ 32767)

송도는 직사각형 모양으로 생긴 도시이다. 두 좌표 사이의 거리는 x 좌표의 차이 + y 좌표의 차이 이다. (맨해튼 거리)

출력

각 테스트 케이스에 대해서 상근이와 친구들이 행복하게 페스티벌에 갈 수 있으면 "happy", 중간에 맥주가 바닥나서 더 이동할 수 없으면 "sad"를 출력한다. 

예제 입력

2 2 0 0 1000 0 1000 1000 2000 1000 2 0 0 1000 0 2000 1000 2000 2000

예제 출력

happy
sad

 

2.  문제 풀이

첫 번째 접근은 x, y 좌표를 하나씩 이동해 가면서 갈 수 있는 거리 인가를 측정하려고 했지만 범위가 너무커 메모리 초과때문에 다른 접근으로 바꿨다.

두 번째 좌표 압축을 고려해 보았다. 50씩 좌표를 압축해서 풀어보려 했지만 골드 5문제 치고 난이도가 너무 높았고 경우의 수도 많은 것 같아 포기 했다

세 번째로 그래프로 생각해 보았다 집, 편의점, 페스티벌 장소 모두 각각의 노드로 보고 각각의 노드에서 서로간의 거리를 저장 해두어 1000이내면 가고 아니면 못가는 식으로 풀었다.

 

3. 코드

from collections import deque

T = int(input())

for _ in range(T):
    n = int(input())
    graph = [[] for _ in range(n+2)]
    location = []
    for i in range(n+2):
        location.append(list(map(int, input().split())))
        for j in range(i):
            graph[j].append((i, abs(location[j][0] - location[i][0]) + abs(location[j][1] - location[i][1])))
            graph[i].append((j, abs(location[j][0] - location[i][0]) + abs(location[j][1] - location[i][1])))

    visited = [0] * (n+2)

    queue = deque()
    queue.append(0)
    visited[0] = 1

    while queue:
        x = queue.popleft()

        for i in graph[x]:
            if i[1] <= 1000 and not visited[i[0]]:
                queue.append(i[0])
                visited[i[0]] = 1

    if visited[n+1]:
        print('happy')
    else:
        print('sad')