V개의 정점, E개의 간선, 들려야할 정점 P가 주어진다. 1부터 V까지의 최단거리와 1부터 P를 들려 V에 도착하는 최단거리가 같을 시 “SAVE HIM”을, 다를 시 “GOOD BYE”를 출력하는 문제. 첫 번째 풀이는 플로이드 와샬 방법을 이용하여 모든 정점으로부터의 최단거리를 계산한 후, 1부터 V까지의 값과 1부터 P까지의 값 + P부터 V까지의 값을 비교하여 풀었으나 시간초과 결과를 받게 되었다. 두 번째 풀이는 INF와 heap을 사용하는 다익스트라를 구현하여 1부터의 최단거리들, P부터의 최단거리를 계산하여 위 공식을 대입해 풀었다.
\n# import sys\n# INF = sys.maxsize\n# input = sys.stdin.readline\n#\n# def floyd(graph):\n# for k in range(1, V+1):\n# graph[k][k] = 0\n# for i in range(1, V+1):\n# for j in range(1, V+1):\n# graph[i][j] = min(graph[i][k] + graph[k][j], graph[i][j])\n#\n#\n# V, E, P = map(int, input().split())\n# graph = [[INF for _ in range(V+1)] for _ in range(V+1)]\n#\n# for _ in range(E):\n# s, e, w = map(int, input().split())\n# graph[s][e] = min(graph[s][e], w)\n# graph[e][s] = min(graph[e][s], w)\n#\n# floyd(graph)\n# print(\"SAVE HIME\" if graph[1][V] == graph[1][P] + graph[P][V] else \"GOOD BYE\")\n\n\nimport heapq\nimport sys\nINF = sys.maxsize\ninput = sys.stdin.readline\n\n\ndef dijkstra(start):\n dist = [INF for _ in range(V+1)]\n dist[start] = 0\n\n q = []\n heapq.heappush(q, [0, start])\n\n while q:\n w, n = heapq.heappop(q)\n\n if dist[n] < w:\n continue\n\n for nn, tw in graph[n]:\n nw = w + tw\n if dist[nn] > nw:\n dist[nn] = nw\n heapq.heappush(q, [nw, nn])\n\n return dist\n\nV, E, P = map(int, input().split())\ngraph = {i: [] for i in range(V+1)}\nfor _ in range(E):\n s, e, w = map(int, input().split())\n graph[s].append([e, w])\n graph[e].append([s, w])\n\n\ndist = dijkstra(1)\ndist_p = dijkstra(P)\nprint(\"SAVE HIM\" if dist[V] == dist[P] + dist_p[V] else \"GOOD BYE\")