‡ CODING TEST STUDY ‡/º 백준

[백준 | Java Silver II ] (#2644) 촌수계산

Trudy | 송연 2024. 7. 11. 21:29

문제

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

 


접근

촌수계산 문제를 읽어보니 깊이를 알아야하는 거니까 dfs 문제임을 알 수 있었다. 

 

graph 크기가 n+1인 2차원 배열이라서 dfs의 for문에서 범위를 n으로 줬는데, n이 현재 접근 중인 정점을 뜻해서 출력이 안되는 이상한 실수를 했다. 

 

그거만 잡아주니 잘 풀렸다!

 

 

최종코드

package week10.baek.july9;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class S2644 {
    static int[][] graph;
    static boolean[] visited;
    static int result = -1;

    public static void dfs(int b, int count, int n){
        visited[n] = true;

        if(n == b) {
            result = count;
            return;
        }

        for (int i = 1; i< graph.length; i++) {
            if(!visited[i] && graph[n][i] == 1){
                dfs( b, count+1, i);
            }
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());

        StringTokenizer st = new StringTokenizer(br.readLine());
        int a = Integer.parseInt(st.nextToken());
        int b = Integer.parseInt(st.nextToken());

        int m = Integer.parseInt(br.readLine());

        graph = new int[n+1][n+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());
            graph[x][y] = 1;
            graph[y][x] = 1;
        }

        visited = new boolean[n+1];
        dfs(b,0, a);

        System.out.println(result);
    }
}