백준 1967 트리의 지름 Java [트리, DFS]

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

1967번: 트리의 지름

이전에 푼 문제와 같은데 설명이 더 상세히 추가됐다

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

public class Main {
    static int[] vtd;
    static int V;
    static HashMap<Integer, ArrayList<int[]>> nodes;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        V = Integer.parseInt(br.readLine());
        if(V == 1) {
            System.out.println(0);
            System.exit(0);
        }
        nodes = new HashMap<>();
        for (int n = 0; n < V-1; n++) {
            int[] arr = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
            setnode(arr[0], arr[1], arr[2]);
            setnode(arr[1], arr[0], arr[2]);
        }
        vtd = new int[V+1];
        Arrays.fill(vtd, -1);
        vtd[1] = 0;
        dfs(1);
        dfs(getmax()[0]);
        System.out.println(getmax()[1]);
    }
    static void setnode(int parent, int child, int len) {
        ArrayList<int[]> value;
        if (nodes.containsKey(parent)) value = nodes.get(parent);
        else value = new ArrayList<>();
        value.add(new int[]{child, len});
        nodes.put(parent, value);
    }
    static void dfs(int node) {
        for (int[] arr : nodes.get(node)) {
            if (vtd[arr[0]] == -1) {
                vtd[arr[0]] = vtd[node] + arr[1];
                dfs(arr[0]);
            }
        }
    }
    static int[] getmax() {
        int index = 1, len = vtd[index];
        for(int i = 2; i <= V; i++) {
            if (vtd[i] > len) {
                index = i;
                len = vtd[i];
            }
        }
        Arrays.fill(vtd, -1);
        vtd[index] = 0;
        return new int[]{index, len};
    }
}

* 문제 풀면서 주의할점이 노드의 개수가 1개일 때가 있다

1개일때는 BufferedReader로 읽을게 없어서 런타임에러(NullPointer)가 발생할 수 있으므로 예외 처리해줬다

뭐가 잘못된지 몰라서 StringTokenizer로 하나씩 잘라보고, Scanner도 써봤는데 같은 에러가 나오길래

문제를 다시읽어보며 1 <= n <= 10000을 확인해 쉽게(?) 해결할 수 있었다