백준 1389 케빈 베이컨의 6단계 법칙

// 16분
 
#define _CRT_SECURE_NO_WARNINGS
#include <cstdio>
#include <algorithm>
using namespace std;
 
const int INF = 1000000000;
 
int main() {
    int n, m, dist[101][101];
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
            dist[i][j] = (i == j ? 0 : INF);
 
    for (int i = 0; i < m; i++) {
        int a, b;
        scanf("%d %d", &a, &b);
        dist[a][b] = dist[b][a] = 1;
    }
 
    for (int k = 1; k <= n; k++)
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= n; j++)
                dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j]);
 
    int num = INF;
    int ans;
    for (int i = 1; i <= n; i++) {
        int sum = 0;
        for (int j = 1; j <= n; j++) {
            sum += dist[i][j];
        }
        if (sum < num) {
            num = sum;
            ans = i;
        }
    }
    printf("%d\n", ans);
    return 0;
}
 

플로이드-와샬 알고리즘 적용.

코드를 좀 더 깔끔하게 짤 수 있을 거 같기도 하고..