본문 바로가기
C++ 알고리즘

백준 1926 : 그림 (BFS)

by hoshi03 2024. 8. 4.

 

문제

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

입력

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

출력

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

예제 입력 1 복사

6 5
1 1 0 1 1
0 1 1 0 0
0 0 0 0 0
1 0 1 1 1
0 0 1 1 1
0 0 1 1 1

예제 출력 1 복사

4
9

 

풀이

 

solution 함수로 방문하지 않은 1인 칸이면 bfs를 시작, 그림 갯수 +1을 해주고

bfs 내부에서는 넓이를 1로 초기화한 후 4방향 탐색하면서 갈 수 있는 칸이면 가면서 넓이를 1씩 더하고 더이상 갈수 없을때 리턴하는 구조다

 

전형적인 bfs 문제라고 생각했는데 c++로 bfs를 오랜만에 풀었더니 자잘한 실수가 엄청 나왓다

s를 solution 메서드에 박아서 크기 측정 불가

arr, visited의 크기를 n,m 입력받기 전에 해서 비정상 종료 

x,y 햇갈려서 입력 for문 범위 문제 등 여러 실수를 했다

앞으로는 아래처럼 y,x로 명시해서 범위 문제가 일어나지 않게 하고

전역 벡터 선언할때 크기를 main에서 resize 해주는걸 까먹지 말자

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

int n, m, res = 0, s= 1, maxS = 0;
vector<vector<int>> arr;
vector<vector<bool>> visited;
int posX[] = {0,1,0,-1};
int posY[] = {1,0,-1,0};
queue<pair<int,int>> q;

void bfs(int y, int x){
    q.emplace(y,x);
    visited[y][x] = true;
    s = 1;
    while (!q.empty()){
        auto tmp = q.front();
        y= tmp.first;
        x = tmp.second;
        q.pop();
        for (int i = 0; i < 4; i++) {
            int dy = y + posY[i];
            int dx = x + posX[i];
            if (dx < 0 || dy < 0 || dx >= m || dy >= n) continue;
            if (visited[dy][dx] || arr[dy][dx] == 0) continue;
            q.emplace(dy,dx);
            visited[dy][dx] = true;
            s++;
        }
    }
}

void solution(){
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (!visited[i][j] && arr[i][j] == 1){
                res++;
                bfs(i,j);
                maxS = max(s, maxS);
            }
        }
    }
}



int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> m; //n 세로 m 가로

    arr.resize(n, vector<int>(m));
    visited.resize(n, vector<bool>(m, false));

    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            int tmp;
            cin >> tmp;
            arr[i][j] = tmp;
        }
    }

    solution();

    int maa = 0;

    cout << res <<'\n';
    cout << maxS << '\n';

}