[백준] 촌수계산 - Silver 2

[jusung-c]

문제 링크

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

문제 조건

• 전체 사람의 수는 n명이며, 사람들은 1, 2, ..., n (1 ≤ n ≤ 100)의 연속된 번호로 표시된다.
• 여러 사람들에 대한 부모 자식들 간의 관계가 주어졌을 때, 주어진 두 사람의 촌수를 계산하라.
• 두 사람의 친척 관계가 전혀 없어 촌수를 계산할 수 없을 때에는 -1을 출력한다.

---

아이디어

1. 친척관계를 전부 리스트로 표현한다.
2. 촌수를 나타내는 dist 배열을 전부 -1로 초기화해둔다.
3. 주어진 두 사람의 번호 중 하나를 시작점으로 잡아 BFS를 진행하면서 촌수를 계산해 dist 배열을 갱신한다.
4. 나머지 번호의 dist 값을 출력한다.

---

시간 복잡도

BFS의 시간 복잡도는 O(V+E)로 정점은 최대 100개 간선은 최대 100+99+...+1 이므로 시간 안 풀이가 가능하다.

---

구현

import java.io.*;
import java.util.*;

public class Main {
    static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));;
    static StringTokenizer st;
    static int N, start, end, M;
    static ArrayList<Integer>[] adj;
    static int[] dist;

    public static void init() throws IOException {
        N = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());
        start = Integer.parseInt(st.nextToken());
        end = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(br.readLine());

        adj = new ArrayList[N + 1];
        for (int i = 1; i <= N; i++) {
            adj[i] = new ArrayList<>();
        }

        dist = new int[N + 1];
        for (int i = 1; i <= N; i++) {
            dist[i] = -1;
        }

        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int x = Integer.parseInt(st.nextToken());
            int y = Integer.parseInt(st.nextToken());

            adj[x].add(y);
            adj[y].add(x);
        }

    }

    private static void pro() throws IOException {
        bfs(start);

        bw.write(dist[end] + " ");
    }

    private static void bfs(int start) {
        Queue<Integer> que = new LinkedList<>();
        que.add(start);
        dist[start] = 0;

        while (!que.isEmpty()) {
            int x = que.poll();

            for (int y : adj[x]) {
                if (dist[y] != -1) continue;

                que.add(y);
                dist[y] = dist[x] + 1;
            }
        }
    }

    public static void main(String[] args) throws Exception {
        init();

        pro();

        bw.close();
    }
}