[BOJ] 7576번 토마토 - C++

"DFS와 BFS"

Posted by Yongmin on July 20, 2021

문제

토마토

풀이

최소날짜를 구하는 문제로, BFS를 이용하여 풀이하였다. 처음 주어진 map에서 1인 지점의 좌표들을 먼저 넣음으로써 시작하였다. 그 이후, 상하좌우에 map이 0이며 방문하지 않으면 큐에 넣는 식을 반복하였다. -1의 경우는 미리 visit을 했다고 셋팅하여 -1을 밟지 못하게 만들어놨다. check함수를 이용하여, 마지막에 작성된 check함수의 값 -1로 정답을 도출해냈다. ++) 토마토, 7579번 문제도 같은 코드를 사용하며, 다만, 3차원을 표현하기 위해 z 축이 생긴점, 이를 위해 make_pair(make_pair<i, j>, k), queue<pair<int, int>, int> q; 를 사용한 점 외에는 똑같다.

소스 코드

** 7576번 **

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include <iostream>
#include <vector>
#include <climits>
#include <queue>

using namespace std;

vector<vector<int>> map(1001, vector<int> (1001));
vector<vector<int>> check(1001, vector<int> (1001));
vector<vector<bool>> visit(1001, vector<bool> (1001));



int N, M, cnt;
int dx[4] = {0, 0, -1, 1};
int dy[4] = {1, -1, 0, 0};

void bfs(){
    queue<pair<int, int>> q;
    
    for(int i = 1; i<=M; i++){
        for(int j = 1; j<=N; j++){
            if(map[i][j] == 1){
                q.push(make_pair(i, j));
                check[i][j] = 1;
                visit[i][j] = true;
            }
            if(map[i][j] == -1){
                check[i][j] = -1;
                visit[i][j] = true;
            }
        }
    }
    
    
    while(!q.empty()){
        cnt++;
        int y = q.front().first;
        int x = q.front().second;
        q.pop();
        
        for(int i = 0; i<4; i++){
            if(y+dy[i] <= M && y+dy[i] >= 1 && x+dx[i] <= N && x+dx[i] >=1 && map[y+dy[i]][x+dx[i]] == 0 && visit[y+dy[i]][x+dx[i]] == false){
                check[y+dy[i]][x+dx[i]] = check[y][x] + 1;
                q.push(make_pair(y+dy[i], x+dx[i]));
                visit[y+dy[i]][x+dx[i]] = true;
            }
        }
    }
    
}

int main(){
    
    scanf("%d %d", &N, &M);
    
    for(int i = 1; i<=M; i++){
        for(int j = 1; j<=N; j++){
            scanf("%d", &map[i][j]);
        }
    }
    
    bfs();
    
    for(int i = 1; i<=M; i++){
        for(int j = 1; j<=N; j++){
            if(check[i][j] == 0){
                printf("%d\n", -1);
                return 0;
            }
        }
    }
    
    int result = 0;
    for(int i = 1; i<=M; i++){
        for(int j = 1; j<=N; j++){
            result = max(result, check[i][j]);
        }
    }
    
    printf("%d\n", result-1);
    
}

** 7579번 **

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#include <iostream>
#include <vector>
#include <climits>
#include <queue>

using namespace std;

vector<vector<vector<int>>> map(101, vector<vector<int>> (101, vector<int> (101)));
vector<vector<vector<int>>> check(101, vector<vector<int>> (101, vector<int> (101)));
vector<vector<vector<bool>>> visit(101, vector<vector<bool>> (101, vector<bool> (101)));



int N, M, H, cnt;
int dx[6] = {0, 0, -1, 1, 0, 0};
int dy[6] = {1, -1, 0, 0, 0, 0};
int dz[6] = {0, 0, 0, 0, 1, -1};

void bfs(){
    queue<pair<pair<int, int>, int>> q;
    
    for(int i = 1; i<=M; i++){
        for(int j = 1; j<=N; j++){
            for(int k = 1; k<=H; k++){
                if(map[i][j][k] == 1){
                    q.push(make_pair(make_pair(i, j), k));
                    check[i][j][k] = 1;
                    visit[i][j][k] = true;
                }
                if(map[i][j][k] == -1){
                    check[i][j][k] = -1;
                    visit[i][j][k] = true;
                }
            }
            
        }
    }
    
    
    while(!q.empty()){
        cnt++;
        int y = q.front().first.first;
        int x = q.front().first.second;
        int z = q.front().second;
        q.pop();
        
        for(int i = 0; i<6; i++){
            if(y+dy[i] <= M && y+dy[i] >= 1 && x+dx[i] <= N && x+dx[i] >=1 && z+dz[i] <= H && z+dz[i] >=1 && map[y+dy[i]][x+dx[i]][z+dz[i]] == 0 && visit[y+dy[i]][x+dx[i]][z+dz[i]] == false){
                check[y+dy[i]][x+dx[i]][z+dz[i]] = check[y][x][z] + 1;
                q.push(make_pair(make_pair(y+dy[i], x+dx[i]), z+dz[i]));
                visit[y+dy[i]][x+dx[i]][z+dz[i]] = true;
            }
        }
    }
    
}

int main(){
    
    scanf("%d %d %d", &N, &M, &H);
    for(int k = 1; k<=H; k++){
        for(int i = 1; i<=M; i++){
            for(int j = 1; j<=N; j++){
                    scanf("%d", &map[i][j][k]);
                }
        }
    }
    
    bfs();
    
    for(int k = 1; k<=H; k++){
        for(int i = 1; i<=M; i++){
            for(int j = 1; j<=N; j++){
                    if(check[i][j][k] == 0){
                        printf("%d\n", -1);
                        return 0;
                    }
            }
        }
    }
    
    int result = 0;
    for(int k = 1; k<=H; k++){
        for(int i = 1; i<=M; i++){
            for(int j = 1; j<=N; j++){
                    result = max(result, check[i][j][k]);
            }
        }
    }
    
    printf("%d\n", result-1);
    
}


# # #