백준 1926번 C++ 풀이

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

1926번: 그림

시간 제한	메모리 제한	solved.ac 티어
2초	128MB	실버 1

문제

어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라. 단, 그림이라는 것은 1로 연결된 것을 한 그림이라고 정의하자. 가로나 세로로 연결된 것은 연결이 된 것이고 대각선으로 연결이 된 것은 떨어진 그림이다. 그림의 넓이란 그림에 포함된 1의 개수이다.

입력

첫째 줄에 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다. 두 번째 줄부터 n+1 줄 까지 그림의 정보가 주어진다. (단 그림의 정보는 0과 1이 공백을 두고 주어지며, 0은 색칠이 안된 부분, 1은 색칠이 된 부분을 의미한다)

출력

첫째 줄에는 그림의 개수, 둘째 줄에는 그 중 가장 넓은 그림의 넓이를 출력하여라. 단, 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이다.

---

전형적인 그래프 탐색 문제이다. 최대 넓이를 저장할때 초기화하는 값에 주의해야 한다. 필자는 습관처럼 -1로 적었다가 계속 틀렸다.

 

필자의 코드는 그래프의 좌상단부터 우하단까지 순차적으로 보면서 1이 나오면 bfs를 실행하고, 연결된 그림을 전부 방문처리하는 방식으로 처리했다. 그런데 만약 그림이 없다면 최대 넓이를 출력할 때, 초기화한 -1이 출력되고, 오답이 된다(53%에서 틀리더라).

 

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;

int n, m;
vector<vector<int>> graph;

void input() {
    cin >> n >> m;
    graph.resize(n, vector<int>(m));
    for (auto& row : graph) {
        for (int& x : row) cin >> x;
    }
}

int bfs(int st_x, int st_y) {
    queue<pair<int, int>> q;
    int dx[4] = {1, 0, -1, 0};
    int dy[4] = {0, 1, 0, -1};
    int area = 0;
    
    q.push({st_x, st_y});
    graph[st_y][st_x] = -1; // 방문 처리
    area++;

    while(!q.empty()) {
        auto [cur_x, cur_y] = q.front();
        q.pop();

        for(int i = 0; i < 4; i++) {
            int next_x = cur_x + dx[i];
            int next_y = cur_y + dy[i];
            if(next_x < 0 || next_x >= m || next_y < 0 || next_y >= n) continue;
            
            if (graph[next_y][next_x] == 1) {
                graph[next_y][next_x] = -1;
                q.push({next_x, next_y});
                area++;
            }
        }
    }

    return area;
}

void sol() {
    int cnt = 0;
    int max_area = 0;
 
    for(int y = 0; y < n; y++) {
        for(int x = 0; x < m; x++) {
            if (graph[y][x] == 1) {
                cnt++;
                max_area = max(max_area, bfs(x, y));
            }
        }
    }
    
    cout << cnt << endl << max_area;
}

int main () {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    input();
    sol();
}