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



하나의 정점에 대해서 연관된 정점을 모두 들어가서 1로 바꿔주는 문제이다.

문제를 보면 dfs 또는 bfs로 풀 수 있는데, 연관된 정점을 모두 구하는 방식의 차이인 것 같다.



큐에 담아서 모두 연결시켜줄 수 도 있지만,
나는 그냥 dfs 함수를 하나 구현해서 첫 번째 노드와 연결되어 있고 이 전에 방문한 적이 없으면 visited를 1로 바꿔주었다.


그리고 다시 2번째 노드를 검사할때는 visited를 0으로 초기화해줘서 다시 연결되어 있는 노드를 모두 검사하는 방식으로 문제를 해결했다.




1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include <iostream>
#include <string.h>
using namespace std;
 
int arr[100][100]={0};
int visited[100]={0};
int n;
 
void dfs(int j){
    for(int k=0;k<n;k++){
        if(arr[j][k] && !visited[k]){
            visited[k]++;
            dfs(k);
        }
    }
}
int main(){
    
    cin>>n;
    
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            scanf("%d",&arr[i][j]);
    
    for(int i=0;i<n;i++){
        memset(visited,0,sizeof(visited));
        dfs(i);
        
        for(int j=0;j<n;j++){
            if(visited[j])
                arr[i][j]=1;
        }
    }
    
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++)
            cout<<arr[i][j]<<" ";
        
        cout<<endl;
    }
    return 0;
}
cs


'알고리즘(BOJ) > DFS' 카테고리의 다른 글

백준 1987번 - 알파벳  (0) 2018.01.28
백준 2468번 - 안전 영역  (3) 2018.01.28
백준 2583번 - 영역 구하기  (2) 2018.01.28
백준 2667번 - 단지번호붙이기  (0) 2018.01.28
백준 1012번 - 유기농 배추  (0) 2018.01.28

+ Recent posts