[백준 알고리즘][DFS] 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으로 출력

풀이

이전에 들어왔던 input 이랑 다르게 인접행렬 자체가 들어오므로 i, j를 입력받으면서 인접리스트를 만들고 n번 for문을 돌면서 DFS를 하고, i번째 DFS를 했을 때 방문할 수 있다면 output 그래프를 1로 바꿔준다

코드

#include<stdio.h>

#include<iostream>

#include<vector>

using namespace std;

class Graph{

public:

int N; 

int results[101][101];

vector<vector<int>> adj;

vector<int> visited; 

Graph(){};

Graph(int n)

{

int i;

N=n;

adj.resize(n);

visited.resize(n);

for(i=0; i<n; i++)

fill_n(results[i], n, 0);

}

void makeEdge(int u, int v)

{

adj[u].push_back(v);

}

void init()

{

visited.clear(); //초기화 처음에 vector<bool> visited로 만들어서 잘 안됐음

visited.assign(N,0);

}

void result(int j)

{

int i, cnt;

for(i=1; i<visited.size(); i++){

if(visited[i] == 1){

results[j][i]=1;

}

}

}

void DFS(int i)

{

if(visited[i] == 0)

visited[i]=1;

vector<int>::const_iterator next;

for(next = adj[i].begin() ; next != adj[i].end() ; next++)

{

if(visited[*next] == 0) //2일때는 출발점

DFS(*next);

else if(visited[*next] == 2){

visited[*next]=0;

DFS(*next);

}

}

}

};

void main()

{

int n, input;

int i, j;

scanf("%d", &n);

Graph G(n+1);

for(i=1; i<=n; i++)

{

for(j=1; j<=n; j++)

{

scanf("%d", &input);

if(input == 1)

{

G.makeEdge(i, j);

}

}

}

for(i=1; i<=n; i++)

{

G.visited[i]=2;

G.DFS(i);

G.result(i);

G.init();

}

for(i=1; i<=n; i++)

{

for(j=1; j<=n ;j++)

printf("%d ", G.results[i][j]);

printf("\n");

}

}