본문 바로가기

백준

2644 촌수계산

와~~ 딱 보니 그래프 문제였다. dfs로 depth 세면서 해당 노드가 나오면 길이 계산했다. 오랜만의 그래프문제

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <deque>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>
#include <stdio.h>
#include <math.h>
#include <sstream>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using iis = pair<int, string>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;

bool visited[10005];
vector <int> adj[10005];
int ans = -1;
int n, x, y;

void dfs(int curr, int len)
{
    visited[curr] = true;
    // printf("%d %d\n", curr, len);
    
    if (curr == y)
        ans = len;
    
    for (int next : adj[curr])
    {
        if (!visited[next])
            dfs(next, len + 1);
    }
}

int main() 
{
    scanf("%d %d %d", &n, &x, &y);
    
    int m;
    scanf("%d", &m);
    
    for (int i = 0; i < m; i++)
    {
        int u, v;
        scanf("%d %d", &u, &v);
        adj[u].push_back(v);
        adj[v].push_back(u);
    }

    dfs(x, 0);
    
    printf("%d\n", ans);
    
    return 0;
}

'백준' 카테고리의 다른 글

7562 나이트의 이동  (0) 2021.12.21
4307 개미  (0) 2021.12.21
2504 괄호의 값  (0) 2021.12.20
1914 하노이탑  (0) 2021.12.20
1541 잃어버린 괄호  (0) 2021.12.14