[JAVA-자바] 1167번: 트리의 지름

 

소스코드:

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class Main {
    static BufferedReader br;
    static BufferedWriter bw;

    static int farNode;
    static int maxDistance;
    static boolean[] visited;
    static List<int[]>[] graph;

    public static void main(String[] args) throws IOException {
        br = new BufferedReader(new InputStreamReader(System.in));
        bw = new BufferedWriter(new OutputStreamWriter(System.out));

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

        visited = new boolean[V + 1];
        graph = new ArrayList[V + 1];
        for (int i = 1; i <= V; i++) {
            graph[i] = new ArrayList<>();
        }

        for (int i = 1; i <= V; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int node = Integer.parseInt(st.nextToken());
            while (true) {
                int targetNode = Integer.parseInt(st.nextToken());
                if (targetNode == -1) {
                    break;
                }
                int score = Integer.parseInt(st.nextToken());
                graph[node].add(new int[]{targetNode, score});
            }

        }
        br.close();

        farNode = 0;
        maxDistance = 0;
        visited[1] = true;
        dfs(1, 0);
        visited[1] = false;

        maxDistance = 0;
        visited[farNode] = true;
        dfs(farNode, 0);
        visited[farNode] = false;

        bw.write(Integer.toString(maxDistance));
        bw.close();
    }

    private static void dfs(int current, int score) {
        if (maxDistance < score) {
            maxDistance = score;
            farNode = current;
        }

        for (int[] node : graph[current]) {
            int next_idx = node[0];
            int next_score = node[1];
            if (!visited[next_idx]) {
                visited[next_idx] = true;
                dfs(next_idx, score + next_score);
                visited[next_idx] = false;
            }
        }

    }
}

 

글의 내용 중 잘못된 점이나 수정이 필요한 부분, 혹은 궁금한 사항이 있다면 언제든 댓글로 남겨주시면 감사하겠습니다.

여러분의 피드백은 더 나은 글을 작성하는 데 큰 도움이 됩니다. 감사합니다.