CS/Algorithm

[BOJ][C++] 1012 유기농배추

별토끼. 2019. 1. 12. 15:15
반응형

[BOJ][C++] 1012 유기농배추

 

문제

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

 

풀이

DFS를 이용해서 풀었습니다. 배추가 있는 곳을 체크한 후 깊이 탐색을 합니다. 깊이탐색을 시작할 때 한 번 카운팅을 해준 후 인접한 곳의 배추들은 카운팅을 하지 않고 visited를 바꿔줍니다.

visited가 1인 곳은 다시 방문하지 않고 다시 메인함수로 와서 방문하지 않은 다른 배추 위치를 찾습니다. 이를 반복하면 수월하게 풀 수 있습니다.

 

코드

#define _CRT_SECURE_NO_WARNINGS
#include <cstdio>
using namespace std;
int M, N, K;
int map[50][50];
int visited[50][50];
int dx[] = { 1,-1,0,0 };
int dy[] = { 0,0,1,-1 };
void reset() {
    for (int i = 0; i < 50; i++) {
        for (int j = 0; j < 50; j++) {
            map[i][j] = 0;
            visited[i][j] = 0;
        }
    }
}
void CountingWorm(int x, int y) {
    visited[x][y] = 1;
    for (int i = 0; i < 4; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (nx < 0 || ny < 0 || nx >= M || ny >= N) continue;
        if (map[nx][ny] == 1 && visited[nx][ny] == 0) {
            CountingWorm(nx, ny);
        }
    }
}
int main() {
    int T;
    scanf("%d", &T);

    for (int t = 1; t <= T; t++) {
        reset();
        scanf("%d %d %d", &M, &N, &K);
        for (int k = 0; k < K; k++) {
            int tmp1, tmp2;
            scanf("%d %d", &tmp1, &tmp2);
            map[tmp1][tmp2] = 1;
        }
        
        int cnt = 0;
        for (int i = 0; i < M; i++) {
            for (int j = 0; j < N; j++) {
                if (map[i][j] == 1 && visited[i][j] == 0) {
                    cnt++;
                    CountingWorm(i, j);
                }
            }
        }
        printf("%d\n", cnt);
    }
}

 

반응형