[5214번] 환승

문제 출처 :

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

알고리즘 분석 :

문제 해결에 필요한 사항

1. BFS

BFS를 통해서 문제를 해결 할 수 있다.

하이퍼 튜브를 하나의 정점으로 생각하여(10만 + 1 ~ 10만 + M) 그래프를 모델링하면 아래 그림과 같이 모델링 할 수 있다.

그 후 BFS를 돌려 최단 거리를 구하는데 하이퍼 튜브를 지나면 cnt로, 하이퍼 튜브가 아닌 다른 정점을 지나면 cnt + 1을 하여 정답을 구한다.

소스 코드 : 

#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
 
using namespace std;
 
int hyper[1002];
vector<int> adj[101111];
bool visit[101111];
 
int main()
{
    int n, k, m;
    scanf("%d %d %d", &n, &k, &m);
 
    for (int i = 1; i <= m; i++)
    {
        for (int j = 0; j < k; j++)
            scanf("%d", &hyper[j]);
 
        for (int j = 0; j < k; j++)
        {
            adj[100000+i].push_back(hyper[j]);
            adj[hyper[j]].push_back(100000+i);
        }
    }
 
    queue<pair<int, int> > q;
 
    q.push({ 1, 1 });
 
    int ans = 0;
    while (!q.empty())
    {
        int here = q.front().first;
        int cnt = q.front().second;
 
        q.pop();
        if (visit[here])
            continue;
 
        visit[here] = true;
 
        int len = adj[here].size();
        for (int i = 0; i < len; i++)
        {
            int there = adj[here][i];
            if (there == n)
            {
                printf("%d", cnt + 1);
                return 0;
            }
 
            if (there > 100000)
                q.push({ there, cnt });
            else
                q.push({ there, cnt + 1 });
        }
    }
 
    printf("-1");
 
    return 0;
}
 
//                                                       This source code Copyright belongs to Crocus
//                                                        If you want to see more? click here >>