n개의 노드, m개의 쌍방향 간선이 있을 때 최단거리로 다른 모든 노드들을 방문할 때 먼저 들리는 노드의 번호를 모든 노드에 대해서 출력하는 문제. 3중 반복문을 사용하는 플로이드 와샬 방법을 이용하여 풀었다. ik + kj 값으로 거리가 갱신될 시 첫번째로 방문하는 노드를 저장하는 배열의 값 ik를 ij에 저장하여 풀었다. 두 번째 풀이는 숏코딩으로 풀었다.
\nimport sys\nINF = sys.maxsize\ninput = sys.stdin.readline\n\ndef solve():\n dist = graph\n route = [[i for i in range(n+1)] for _ in range(n+1)]\n\n for k in range(1, n+1):\n dist[k][k] = 0\n route[k][k] = \"-\"\n\n for i in range(1, n+1):\n for j in range(1, n+1):\n next = graph[i][k] + graph[k][j]\n if dist[i][j] > next:\n dist[i][j] = next\n route[i][j] = route[i][k]\n\n return route\n\nn, m = map(int, input().split())\ngraph = [[INF for _ in range(n+1)] for _ in range(n+1)]\n\nfor _ in range(m):\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\nfor r in solve()[1:]:\n print(\" \".join(map(lambda x: str(x), r[1:])))\n\n\n# import sys\n# INF = sys.maxsize\n# r = sys.stdin.readline\n#\n# n, m = map(int, r().split())\n# g = [[INF] * (n+1) for _ in range(n+1)]\n# ro = [[i for i in range(n+1)] for _ in range(n+1)]\n#\n# for _ in range(m):\n# s, e, w = map(int, r().split())\n# g[s][e] = min(g[s][e], w)\n# g[e][s] = min(g[e][s], w)\n#\n#\n# for k in range(1, n+1):\n# g[k][k] = 0\n# ro[k][k] = \"-\"\n# for i in range(1, n+1):\n# for j in range(1, n+1):\n# ne = g[i][k] + g[k][j]\n# if g[i][j] > ne:\n# g[i][j] = ne\n# ro[i][j] = ro[i][k]\n#\n# for s in ro[1:]:\n# print(\" \".join(map(lambda x: str(x), s[1:])))