Develope Story

2019-03-22 [JAVA]백준 11403번 - 경로 찾기

22 Mar 2019 | java algorithm_study

문제 출처

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

문제

문제

가중치 없는 방향 그래프 G가 주어졌을 때, 모든 정점 (i, j)에 대해서, i에서 j로 가는 경로가 있는지 없는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 정점의 개수 N (1 ≤ N ≤ 100)이 주어진다. 둘째 줄부터 N개 줄에는 그래프의 인접 행렬이 주어진다. i번째 줄의 j번째 숫자가 1인 경우에는 i에서 j로 가는 간선이 존재한다는 뜻이고, 0인 경우는 없다는 뜻이다. i번째 줄의 i번째 숫자는 항상 0이다.

출력

총 N개의 줄에 걸쳐서 문제의 정답을 인접행렬 형식으로 출력한다. 정점 i에서 j로 가는 경로가 있으면 i번째 줄의 j번째 숫자를 1로, 없으면 0으로 출력해야 한다

문제 풀이

  • 우선 그래프를 나타내는 배열 생성

    for(int i=0; i<N; i++)
            for (int j=0; j<N; j++)
                graph[i][j] = sc.nextInt();

  • bfs 메소드 생성

    public static void bfs(int graph_x, int graph_y){
        isVisited[graph_y] = false;
        Queue<Integer> q = new LinkedList<>();
        q.offer(graph_y);
  
        while (!q.isEmpty()){
            int tmp = q.poll();
  
            for (int i=0; i<N; i++)
                if(isVisited[i]== false && graph[tmp][i] == 1){
                    q.offer(i);
                    graph[graph_x][i] =1;//경로가 있을 경우 해당 그래플 배열을 1로 바꾼다.
                    isVisited[i] = true;
                }
        }
    }
    • 1일때 해당 정점에 경로가 있을경우 queue에 넣고 큐에서 하나씩 꺼내어 해당 정점에 경로가 있을경우 다시 queue에 넣는다. (정점 방문 체크)
    • 경로가 있을 경우 해당 그래플 배열을 1로 바꾼다.

전체 소스

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class Baekjoon_11403 {
    static int[][] graph;
    static boolean[] isVisited;
    static int N;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

         N= sc.nextInt();
        graph = new int[N][N];

        for(int i=0; i<N; i++){ //입력
            for (int j=0; j<N; j++){
                graph[i][j] = sc.nextInt();
            }
        }

        for(int i=0; i<N; i++){ //bfs
            isVisited = new boolean[N]; //방문 체크 초기화
            for (int j=0; j<N; j++){
                if(graph[i][j] == 1 && isVisited[j] == false){
                    bfs(i, j);
                }
            }
        }

        for(int i=0; i<N; i++){ //출력
            for(int j=0; j<N; j++){
                System.out.print(graph[i][j]+" ");
            }
            System.out.println();
        }

    }
    public static void bfs(int graph_x, int graph_y){
        isVisited[graph_y] = false;
        Queue<Integer> q = new LinkedList<>();
        q.offer(graph_y);

        while (!q.isEmpty()){
            int tmp = q.poll();

            for (int i=0; i<N; i++)
                if(isVisited[i]== false && graph[tmp][i] == 1){
                    q.offer(i);
                    graph[graph_x][i] =1; //경로 존재시 1로 변경
                    isVisited[i] = true;
                }
        }
    }
}

Related Posts

Comments