[BOJ 2644] 촌수계산

문제 출처 : https://www.acmicpc.net/problem/2644

 

풀이 :

 

두 사람 사이의 촌수를 계산하는 문제입니다.

이는 두 사람 사이에 엣지가 몇개 있는지를 묻는 문제로 지나간 엣지 갯수를 저장해주면서 탐색하시면 됩니다.

 

#include<stdio.h>
#include<queue>
using namespace std;
 
queue<pair<int,int>>q;
pair<int, int> p;
int n, s, e, map[101][101], check[101], a, b, k;
 
int main()
{
    scanf("%d\n%d %d\n%d", &n, &s, &e, &k);
    while (k--) {
        scanf("%d %d", &a, &b);
        map[a][b] = map[b][a] = 1;
    }
    q.push({ s,0 });
    check[s] = 1;
    while (!q.empty()) {
        p = q.front();
        q.pop();
        for (int i = 1; i <= n; i++) {
            if (map[p.first][i] && !check[i]) {
                if (i == e) {
                    printf("%d", p.second + 1);
                    return 0;
                }
                q.push({ i, p.second + 1 });
                check[i] = 1;
            }
        }
    }
    printf("-1");
}