백준 [11403] 경로 찾기

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

 

 

문제에서 방향 그래프가 주어진다고 했다. 그리고 i -> j 로 가는 길이가 양수인 경로의 뜻은 방향 반대가 아니라 i -> j 방향 그대로 가는 방향을 말하는 것으로 이해했다.

 

풀이

 

필요한 변수들을 우선 생각해보았다.

• 입력 정보를 담을 배열
• 출력할 정보를 담을 배열
• 방문 배열

그 다음 입력받은 배열을 탐색하면서 방문가능한 경우

해당 노드 정보를 dfs로 탐색하면서 양수 방향 경로를 모두 출력 배열에 체크 해주면 되었다.

그래서 dfs 탐색을 돌기 전 방문배열을 초기화 해주면서 해당 행이 양수로 탐색할 수 있는 모든 정보를 탬색하는 것을 반복하면 된다.

 

정답 코드

package com.baekjoon.p11403;

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

public class Main {
    static int N;
    static boolean[] v;
    static int[][] map, ans;

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

        N = Integer.parseInt(br.readLine());

        map = ans = new int[N][N];
        v = new boolean[N];
        StringTokenizer st;

        // 초기화
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < N; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        // 탐색
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (map[i][j] == 1) {
                    v = new boolean[N];
                    dfs(i, j);
                }
            }
        }

        // 출력
        StringBuilder sb = new StringBuilder();
        // sb.append("\n");
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                sb.append(ans[i][j] + " ");
            }
            sb.append("\n");
        }

        System.out.print(sb);
    }

    static void dfs(int r, int c) {
        ans[r][c] = 1;
        v[c] = true;

        for (int i = 0; i < N; i++) {
            if (!v[i] && map[c][i] == 1) {
                ans[r][i] = 1;
                dfs(r, i);
            }
        }
    }
}