[백준 알고리즘] 2799번: 블라인드

봄이 오고 있다. 해는 높이 떠서 환하게 빛나고 있다. 사람들은 햇볕을 가리기 위해 블라인드를 내린다. 상근이는 이웃들이 무엇을 하는지를 염탐하고, 이것에 대해서 뒷담화를 하는 주부이다. 올해는 건너편 아파트에 사는 사람들이 블라인드를 얼마나 내리는지를 조사하려고 한다. 모든 창문은 4X4 그리드로 나타낼 수 있고, 별문자를 이용해서 블라인드를 나타낸다. 상근이가 볼 수 있는 창문은 다음 5가지 상태 중 하나이다. 건너편 아파트의 한 층에는 N개의 창문이 있고, 총 M층 건물이다. 현재 건너편 아파트의 창문 상태가 주어졌을 때, 위의 5가지 상태가 각각 몇 번 나오는지 구하는 프로그램을 작성하시오.

#include <iostream>
#include <vector>
#include <string>

using namespace std;

int arr[5];
vector<string> v;

int main(void){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int M, N;
    cin >> M >> N;

    v.resize(5 * M + 1); // 벡터 크기를 입력된 크기로 재조정
    for (int i = 0; i < 5 * M + 1; i++)
        cin >> v[i];
        for (int i = 0; i < 5 * M + 1; i++){
            bool checked = false; // 창문 윗줄 판정
            for (int j = 0; j < v[i].size(); j++){
                if (v[i][j] == '.'){
                    checked = true;
                    arr[0]++;
                    j += 3;
                }
                else if (v[i][j] == '*'){
                    checked = true;
                    int k = 0;
                    while (1){
                        k++;
                        if (v[i + k][j] != '*') break;
                    }
                    arr[k]++;
                    j += 3;
                    }
                 }
                 if (checked) i += 3;
        }

        for (int i = 0; i < 5; i++)

                 cout << arr[i] << " ";

        cout << "\n";

        return 0;
}