inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

2주차 개념 #9. 깊이우선탐색(DFS, Depth First Search)

선생님 2주차 개념 #9의 예제 문제 질문있습니다.

해결된 질문

333

송민규

작성한 질문수 5

0

해당 코드는 제가 직접 구현한 문제 해답 코드인데요. 5 51 0 1 0 11 1 0 0 10 0 0 1 10 0 0 1 10 1 0 0 0의 테스트 케이스를 통과하도록 짜봤습니다.n * m 의 인접행렬을 만들고 1을 발견할 시 그 위치에서부터 ↑→↓←의 순으로 탐색을 하며 네 방향의 탐색을 마쳤을 때 결과가 모두 1이 아니라면 결과값을 1씩 증가시켜 답을 출력하는 코드입니다.첫번째 질문으로는 제가 구현한 코드가 너무 비효율적인 것 같습니다. 어떻게하면 이 코드에서 더 좋은 코드로 발전시킬 수 있을까요?두번째 질문으로는 제가 구현한 방식을 dfs라고 할 수 있는걸까요? 그냥 감으로 작성한 코드라서 이게 될줄 몰랐습니다..코드는 다음과 같습니다.단순히 구현만 하는것으로는 정답이겠지만 만약 이런 문제가 나온다면 시간초과나 메모리 초과가 날 게 분명해 보입니다../* 연결된 육지는 모두 오염 바다로는 갈 수 없다. 오염시킬수 있는 방향은 상 하 좌 우 네방향 최소 몇번을 이동해야 모든 육지를 오염시킬 수 있는가? 1 : 육지 0 : 바다 범위 1 <= N <= 100 1 <= M <= 100 예제 입력 입력은 맵의 세로길이 N과 가로길이 M이 주어진다. N * M의 맵이 주어진다. 5 5 1 0 1 0 1 1 1 0 0 1 0 0 0 1 1 0 0 0 1 1 0 1 0 0 0 출력 4 */ #include <bits/stdc++.h> using namespace std; const int dy[] = {-1, 0, 1, 0}; const int dx[] = {0, 1, 0, -1}; int n, m, findCount = 0; void dfs(int y, int x, int **ajm){ ajm[y][x] = 0; int check = 0; for(int i = 0; i < 4; i++){ int ny = y + dy[i]; int nx = x + dx[i]; if(ny < 0 || ny >= n || nx < 0 || nx >= n) { check++; if(check == 4){ findCount++; return; } continue;} if(ajm[ny][nx] != 1) { check++; if(check == 4){ findCount++; return; } continue; } if(ajm[ny][nx] == 1){ dfs(ny,nx,ajm); return; } } } int main(){ int num; cin >> n >> m; int **ajm = new int*[n]; for(int i = 0; i < n; i++){ ajm[i] = new int[m]; } for(int i = 0; i < n; i++){ for(int j= 0; j < m; j++){ cin >> ajm[i][j]; } } for(int i = 0; i < n; i++){ for(int j= 0; j < m; j++){ if(ajm[i][j] == 1){ dfs(i, j, ajm); } } } cout << findCount; return 0; }

c++ 코딩-테스트

답변 2

0

큰돌

안녕하세요 민규님 ㅎㅎ

 

첫번째 질문으로는 제가 구현한 코드가 너무 비효율적인 것 같습니다. 어떻게하면 이 코드에서 더 좋은 코드로 발전시킬 수 있을까요?

 

>> 코드리뷰를 드리자면요.

  1. 이게 보시면. 4방향을 탐색할 때 return하는 부분이 너무 비효율적입니다.

예를 들어 왼쪽을 탐색하고 해당 for문을 아예 종료를 시켜버립니다.

즉, 이 코드는 4방향 탐색이라고도 할 수 없습니다. 한방향에서 1이라면 dfs돌리고 return 하거든요.

        if(ajm[ny][nx] == 1){
            dfs(ny,nx,ajm);
            return;
        }

그렇다면 결국...

나머지 방향들은 처음 시작 이중 for 문에 의존하여 실행되게 되는 것이죠.

    for(int i = 0; i < n; i++){
        for(int j= 0; j < m; j++){
            if(ajm[i][j] == 1){
                dfs(i, j, ajm);
            }
        }
    }

지금 문제야... 디버깅코드를 찍어보시면.

start
0 : 4
1 : 4
2 : 4
3 : 4
3 : 3
2 : 3

이런식으로 한방향을 기준으로 탐색하다가... -> -> 다시 되는 방향 찾아가지고... (아래로) 이런식으로 됩니다.

그러나 이런 코드는.

image

이렇게 탐색해야할 때 아래쪽을 전혀 탐색을 못하게 되겠죠? 이런 단점이 있습니다.

참고 : 디버깅 코드.

#include <bits/stdc++.h>
using namespace std;

const int dy[] = {-1, 0, 1, 0};
const int dx[] = {0, 1, 0, -1};
int n, m, findCount = 0;

void dfs(int y, int x, int **ajm){
    cout << y << " : " << x << "\n";
    ajm[y][x] = 0;
    int check = 0;
    for(int i = 0; i < 4; i++){
        int ny = y + dy[i];
        int nx = x + dx[i];
        if(ny < 0 || ny >= n || nx < 0 || nx >= n) {
            check++;
            if(check == 4){
                findCount++;
                return;
            }
            continue;}
        if(ajm[ny][nx] != 1) {
            check++;
            if(check == 4){
                findCount++;
                return;
            }
            continue;
        }
        if(ajm[ny][nx] == 1){
            dfs(ny,nx,ajm);
            return;
        }
    }
}
int main(){
    int num;
    cin >> n >> m;
    int **ajm = new int*[n];
    for(int i = 0; i < n; i++){
        ajm[i] = new int[m];
    }

    for(int i = 0; i < n; i++){
        for(int j= 0; j < m; j++){
            cin >> ajm[i][j];
        }
    }
    for(int i = 0; i < n; i++){
        for(int j= 0; j < m; j++){
            if(ajm[i][j] == 1){
                cout << "start\n";
                dfs(i, j, ajm);
            }
        }
    }
    cout << findCount;
    return 0;
}

 

  1. 원본배열은 왠만하면 수정하지 않는게 좋습니다.


void dfs(int y, int x, int **ajm){
    ajm[y][x] = 0;

지금이야 0으로 바꿔도 괜찮다고 생각합니다. 다만 문제가 복잡해지면.. 원본배열 수정했는데 또 꼬이는 경우가 많이 있어서 visited배열을 따로 두는게 더 좋을 것 같습니다.

 

 

두번째 질문으로는 제가 구현한 방식을 dfs라고 할 수 있는걸까요? 그냥 감으로 작성한 코드라서 이게 될줄 몰랐습니다.

>>

네 dfs가 맞습니다. 재귀적으로 호출하고 있는 부분도 있고 ajm를 수정하면서 가기 때문에 종료도 잘 됩니다.

 




또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.


0

송민규

/*
연결된 육지는 모두 오염
바다로는 갈 수 없다.
오염시킬수 있는 방향은 상 하 좌 우 네방향
최소 몇번을 이동해야 모든 육지를 오염시킬 수 있는가?
1 : 육지 0 : 바다

범위 
1 <= N <= 100
1 <= M <= 100

예제 입력

입력은 맵의 세로길이 N과 가로길이 M이 주어진다. N * M의 맵이 주어진다.
5 5
1 0 1 0 1
1 1 0 0 1
0 0 0 1 1
0 0 0 1 1
0 1 0 0 0

출력
4
*/
#include <bits/stdc++.h>
using namespace std;

const int dy[] = {-1, 0, 1, 0};
const int dx[] = {0, 1, 0, -1};
int n, m, findCount = 0;

void dfs(int y, int x, int **ajm){
    ajm[y][x] = 0;
    int check = 0;
    for(int i = 0; i < 4; i++){
        int ny = y + dy[i];
        int nx = x + dx[i];
        if(ny < 0 || ny >= n || nx < 0 || nx >= n) {
            check++;
            if(check == 4){
                findCount++;
                return;
            }
            continue;}
        if(ajm[ny][nx] != 1) {
            check++;
            if(check == 4){
                findCount++;
                return;
            }
            continue;
        }
        if(ajm[ny][nx] == 1){
            dfs(ny,nx,ajm);
            return;
        }
    }
}
int main(){
    int num;
    cin >> n >> m;
    int **ajm = new int*[n];
    for(int i = 0; i < n; i++){
        ajm[i] = new int[m];
    }

    for(int i = 0; i < n; i++){
        for(int j= 0; j < m; j++){
            cin >> ajm[i][j];
        }
    }
    for(int i = 0; i < n; i++){
        for(int j= 0; j < m; j++){
            if(ajm[i][j] == 1){
                dfs(i, j, ajm);
            }
        }
    }
    cout << findCount;
    return 0;
}

0

송민규

질문글이 이상하게 올라가서 다시 올립니다 ㅠ

2주차 개념#12 트리 순회

0

16

2

백준사이트가 종료된다고 합니다.

0

220

2

백준 서비스 종료

9

701

1

sk 하이닉스 코테 대비

0

350

2

3-G 최댓값 질문

0

48

1

모듈러 연산 값이 10이 아닌 경우도 있지 않나요?

0

82

2

3-I 코드 질문드립니다.

0

61

2

3-N 질문 있습니다.

0

65

2

학습방법

0

100

2

4-H 질문 있습니다 (코드 리뷰)

0

66

2

코딩테스트 어디까지 준비해야 하는지 질문이 있습니다.

0

163

2

2-O 반례가 무엇일지 어떤 부분이 틀렸는지 잘 모르겠습니다.

0

69

2

2주차 개념 #4-2. 인접행렬 질문있습니다.

0

63

2

1-A 문제풀이 후 궁금한 점이 생겨서 질문드립니다.

0

49

2

조합 재귀 풀이 확인 해주시면 감사하겠습니다.

0

67

2

함수별 시간복잡도

0

72

2

3-h 질문입니다.

0

48

1

안녕하세요 선생님. 시간 복잡도 4번 질문있습니다.

0

52

2

1-I 문제 질문 드립니다.

0

76

2

2-P 질문입니다.

0

56

1

mac에서 시작하기 관련

0

87

2

5-Q 질문

0

62

2

풀이 코드 질문

0

62

2

맞왜틀

0

68

2