BOJ 11780 - 플로이드2

플로이드 와샬 알고리즘을 사용하는 문제.

 

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
typedef unsigned long long ll;
const int INF = 1e9;

int dist[101][101];
int via[101][101];
int ks[101][101];

void reconstruct(int i, int j, vector<int>& path) {
	//i->j 경유없이 가는 경우면
	if (via[i][j] == -1) {
		path.push_back(i+1);
		path.push_back(j+1);
		return;
	}

	//i->j 경유점이 있다면
	int w = via[i][j]; //w에 경유점 담아주고
	reconstruct(i, w, path);
	path.pop_back();
	reconstruct(w, j, path);

}

int main() {
	int n, m; scanf("%d %d", &n, &m);
	
	fill(&via[0][0], &via[100][101], -1);
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			dist[i][j] = i == j ? 0 : INF;
		}
	}
	for (int i = 0,u,v,w; i < m; i++) {
		scanf("%d %d %d", &u, &v, &w);
		dist[u-1][v-1] = min(dist[u - 1][v - 1], w);
	}
	for (int k = 0; k < n; k++) {
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				if (dist[i][j] > dist[i][k] + dist[k][j]) {
					dist[i][j] = dist[i][k] + dist[k][j];
					via[i][j] = k;  //유의
				}
			}
		}
	}

	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			printf("%d ", dist[i][j]);
		}
		printf("\n");
	}
			
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			if (dist[i][j] >= INF || dist[i][j]==0) {
				printf("0\n");
				continue;
			}
			vector<int> path;
			reconstruct(i, j, path);
			printf("%d ", path.size());
			for (int num : path) printf("%d ", num);
			printf("\n");
		}
	}
		

}

 점 i~j까지의 거리를 구하는 것은 간단하지만 경로를 띄우는 것이 너무 어려워서 다른 사람의 코드를 참고했던 것 같다.

 

via[][]를 이용해서 경로를 저장하고 reconstruct  함수에서 처리한다.