11779번

11779번 - 최소 비용 구하기 2

1916번과 동일한 문제인데, 최단 거리에 이르는 경로와 그 경로에서 정점의 갯수까지 구하는 문제이다.

BFS, 다익스트라, 벨만 포드는 모두 parent 배열을 이용해서 경로를 구할 수 있다.

distance 값이 바뀔 때, parent[next]=now 로 바꿔주면 되기 때문이다.

이를 이용해서 , 경로 상에서 정점의 갯수와 순서까지 알아낼 수 있다.

이를 order 배열을 이용하거나 stack을 이용해서 구할 수 있다.

<정답 코드>

#include<iostream>
#include<queue>
#include<vector>
 
using namespace std;
const int INF = 987654321;
int n, m, start, dest;
 
int main()
{
 
    scanf("%d %d", &n,&m);
 
    vector < vector<pair<int, int>>> v(n + 1);
    vector<int> dist(n + 1);
    vector<bool> visited(n + 1);
    vector<int> parent(n + 1);
 
    for (int i = 1; i <= n; i++)
    {
        dist[i] = INF;
        visited[i] = false;
        parent[i] = -1;
    }
 
    for (int i = 0; i < m; i++)
    {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        v[a].push_back({ b,c });
    }
 
    scanf("%d %d", &start, &dest);
 
    dist[start] = 0;
 
    while (true)
    {
        int now;
        int distance = INF;
        for (int i = 1; i <= n; i++)
        {
            if (!visited[i] && distance > dist[i] && dist[i] != INF)
            {
                distance = dist[i];
                now = i;
            }
        }
 
        if (distance == INF)
        {
            break;
        }
 
        visited[now] = true;
 
        for (int i = 0; i < v[now].size(); i++)
        {
            int next = v[now][i].first;
            int cost = v[now][i].second;
 
            if (dist[next] > dist[now] + cost)
            {
                dist[next] = dist[now] + cost;
                parent[next] = now;
            }
        }
 
    }
 
    printf("%d\n", dist[dest]);
    vector<int> order;
    int ans = 0;
    int c = dest;
    
    while (c!=-1)
    {
        order.push_back(c);
        ans++;        
        c = parent[c];
        
    }
 
    printf("%d\n", ans);
 
    for (int i = order.size()-1; i >=0; i--)
    {
        printf("%d ", order[i]);
    }
 
    return 0;
 
}