수학과의 좌충우돌 프로그래밍

[C++] BAEKJOON 11404 플로이드 본문

알고리즘/C++

[C++] BAEKJOON 11404 플로이드

ssung.k 2019. 11. 10. 22:17

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

 

11404번: 플로이드

첫째 줄에 도시의 개수 n(1 ≤ n ≤ 100)이 주어지고 둘째 줄에는 버스의 개수 m(1 ≤ m ≤ 100,000)이 주어진다. 그리고 셋째 줄부터 m+2줄까지 다음과 같은 버스의 정보가 주어진다. 먼저 처음에는 그 버스의 출발 도시의 번호가 주어진다. 버스의 정보는 버스의 시작 도시 a, 도착 도시 b, 한 번 타는데 필요한 비용 c로 이루어져 있다. 시작 도시와 도착 도시가 같은 경우는 없다. 비용은 100,000보다 작거나 같은 자연수이다. 시작

www.acmicpc.net

전형적인 플로이드 와샬 알고리즘을 사용하여 해결하는 문제였습니다.

플로이드 와샬 알고리즘이란?

 

[Algorithm] 플로이드 와샬(Floyd-Warshall) 알고리즘

그래프에서 정점끼리의 최단 경로를 구하는 방법은 여러가지가 있습니다. 플로이드 와샬 알고리즘에 대해서 알아보기 전에 이에 대해서 간단히 살펴보도록 하겠습니다. 하나의 정점에서 다른 하나의 정점까지의 최..

ssungkang.tistory.com

#include <iostream>
#include <algorithm>

using namespace std;

#define INF 987654321

int graph[101][101];

void floyd(int n){
    for (int mid=1;mid<=n;mid++){
        for (int start=1;start<=n;start++){
            for (int end=1;end<=n;end++){
                if (graph[start][end] > graph[start][mid] + graph[mid][end])
                    graph[start][end] = graph[start][mid] + graph[mid][end];
            }
        }
    }
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int n,m;
    cin >> n >> m;

    for (int i=1;i<=n;i++){
        for (int j=1;j<=n;j++){
            graph[i][j] = INF;
            if (i==j) graph[i][j] = 0;
        }
    }

    for (int i=0;i<m;i++){
        int node1, node2, w;
        cin >> node1 >> node2 >> w;
        graph[node1][node2] = min(graph[node1][node2],w);
    }

    floyd(n);

    for (int i=1;i<=n;i++){
        for (int j=1;j<=n;j++){
            if (graph[i][j] == INF)
                cout << 0 << " ";
            else
            cout << graph[i][j] << " ";
        }
        cout << "\n";
    }
    
    return 0;
}

 

Comments