본문 바로가기

STUDY/알고리즘

[백준 B1] 성 지키기 (1236번)

www.acmicpc.net/problem/1236

 

1236번: 성 지키기

첫째 줄에 성의 세로 크기 N과 가로 크기 M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에는 성의 상태가 주어진다. 성의 상태는 .은 빈칸, X는 경비원이 있는 칸이다

www.acmicpc.net

문제

영식이는 직사각형 모양의 성을 가지고 있다. 성의 1층은 몇 명의 경비원에 의해서 보호되고 있다. 영식이는 모든 행과 모든 열에 한 명 이상의 경비원이 있으면 좋겠다고 생각했다.

성의 크기와 경비원이 어디있는지 주어졌을 때, 몇 명의 경비원을 최소로 추가해야 영식이를 만족시키는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 성의 세로 크기 N과 가로 크기 M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에는 성의 상태가 주어진다. 성의 상태는 .은 빈칸, X는 경비원이 있는 칸이다.

출력

첫째 줄에 추가해야 하는 경비원의 최솟값을 출력한다.

소스코드 및 풀이

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main(void)
{
    int N, M;
    cin >> N >> M;

    vector< vector<char> > castle(N, vector<char>(M));
    for(int i=0; i<N; ++i) {
        for(int j=0; j<M; ++j) {
            cin >> castle[i][j];
        }
    }

    int row_check[N];
    int col_check[M];

    fill(row_check, row_check + N, 0);
    fill(col_check, col_check + M, 0);

    for(int i=0; i<N; ++i) {
        for(int j=0; j<M; ++j) {
            if(castle[i][j] == 'X') {
                row_check[i] = 1;
                break;
            }
        }
    }

    for(int i=0; i<M; ++i) {
        for(int j=0; j<N; ++j) {
            if(castle[j][i] == 'X') {
                col_check[i] = 1;
                break;
            }
        }
    }

    int row_cnt = 0;
    for(int i=0; i<N; ++i) {
        if(!row_check[i])
            row_cnt++;
    }

    int col_cnt = 0;
    for(int i=0; i<M; ++i) {
        if(!col_check[i])
            col_cnt++;
    }

    int cnt = (row_cnt > col_cnt ? row_cnt  : col_cnt);
    cout << cnt << endl;
    return 0;
}

* 풀고나서 "이게 왜 맞지?"라는 생각이 들었다. 다른 복잡한 알고리즘이 있을 줄 알았는데 그냥 경비원이 필요한 행의 수, 열의 수를 각각 센 후 더 큰 값을 출력해주면 되는 문제이다.