[백준 4803] 트리 (JAVA)

문제

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

풀이

DFS를 통해 탐색하고 이미 방문된 부분이 있으면 사이클이 있는 것으로 간주합니다. (직전 노드는 제외)

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.StringTokenizer;

public class Main {

    static int N, M;
    static ArrayList<Integer>[] tree;
    static boolean[] visited;

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        int T = 1;
        while (true) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            N = Integer.parseInt(st.nextToken());
            M = Integer.parseInt(st.nextToken());
            if (N == 0 && M == 0) break;

            int ans = 0;
            tree = new ArrayList[N];
            visited = new boolean[N];
            for (int i = 0; i < N; i++) {
                tree[i] = new ArrayList<>();
            }

            for (int i = 0; i < M; i++) {
                st = new StringTokenizer(br.readLine());
                int a = Integer.parseInt(st.nextToken()) - 1;
                int b = Integer.parseInt(st.nextToken()) - 1;
                tree[a].add(b);
                tree[b].add(a);
            }

            for (int i = 0; i < N; i++) {
                if (!visited[i]) {
                    visited[i] = true;
                    if (dfs(-1, i)) ans++;
                }
            }

            sb.append("Case ").append(T).append(": ");
            if (ans == 0) {
                sb.append("No trees.\n");
            } else if (ans == 1) {
                sb.append("There is one tree.\n");
            } else {
                sb.append("A forest of ").append(ans).append(" trees.\n");
            }
            T++;
        }

        System.out.println(sb);
        br.close();
    }

    public static boolean dfs(int root, int num) {
        for (int i : tree[num]) {
            if (i == root) continue;
            if (visited[i]) return false;
            visited[i] = true;
            if (!dfs(num, i)) return false;
        }
        return true;
    }
}