[시뮬레이션(테트리스)] (CHECK) 백준 20061 모노미노도미노2 C++

 

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

20061번: 모노미노도미노 2

#include <iostream>

using namespace std;

bool blue[4][6]; // x 기준, y 사라짐
bool green[6][4]; // y 기준, x 사라짐

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

    int n;
    cin >> n;

    int answer = 0;
    
    int t, x, y;
    for(int i=0; i<n; i++) {
        cin >> t >> x >> y;

        // 블록 그대로 이동(비어있는 칸 주의)
        for(int j=0; j<6; j++) {
            if (t == 1) {
                if (j == 5) blue[x][j] = true;
                else if (blue[x][j+1]) blue[x][j] = true;
            }
            else if (t == 2) {
                if (j == 4) {
                    blue[x][j] = true;
                    blue[x][j+1] = true;
                }
                else if (blue[x][j+2]) {
                    blue[x][j] = true;
                    blue[x][j+1] = true;
                }
            }
            else if (t == 3) {
                if (j == 5) {
                    blue[x][j] = true;
                    blue[x+1][j] = true;
                }
                else if (blue[x][j+1] || blue[x+1][j+1]) {
                    blue[x][j] = true;
                    blue[x+1][j] = true;
                }
            }

            if (blue[x][j]) break;
        }

        for(int j=0; j<6; j++) {
            if (t == 1) {
                if (j == 5) green[j][y] = true;
                else if (green[j+1][y]) green[j][y] = true;
            }
            else if (t == 2) {
                if (j == 5) {
                    green[j][y] = true;
                    green[j][y+1] = true;
                }
                else if (green[j+1][y] || green[j+1][y+1]) {
                    green[j][y] = true;
                    green[j][y+1] = true;
                }
            }
            else if (t == 3) {
                if (j == 4) {
                    green[j][y] = true;
                    green[j+1][y] = true;
                }
                else if (green[j+2][y]) {
                    green[j][y] = true;
                    green[j+1][y] = true;
                }
            }

            if (green[j][y]) break;
        }

        // 블록 삭제 및 이동(삭제 후 바로 이동)
        for(int j=0; j<6; j++) {
            if (blue[0][j] && blue[1][j] && blue[2][j] && blue[3][j]) {
                for(int k=0; k<4; k++) {
                    for(int p=j; p>0; p--) {
                        blue[k][p] = blue[k][p-1];
                    }
                }
                answer++;
            }
            if (green[j][0] && green[j][1] && green[j][2] && green[j][3]) {
                for(int k=0; k<4; k++) {
                    for(int p=j; p>0; p--) {
                        green[p][k] = green[p-1][k];
                    }
                }
                answer++;
            }
        }

        // 특별한 칸 확인(바로 삭제 및 이동X)
        int bcnt = 0;
        int gcnt = 0;
        for(int j=0; j<2; j++) {
            if (blue[0][j] || blue[1][j] || blue[2][j] || blue[3][j]) bcnt++;
            if (green[j][0] || green[j][1] || green[j][2] || green[j][3]) gcnt++;
        }

        // 블록 행 이동
        for(int j=0; j<4; j++) {
            if (bcnt == 0) break; // 초기화 방지
            for(int k=5; k>=bcnt; k--) {
                blue[j][k] = blue[j][k-bcnt];
                blue[j][k-bcnt] = false; // 초기화
            }
        }
        for(int j=0; j<4; j++) {
            if (gcnt == 0) break;
            for(int k=5; k>=gcnt; k--) {
                green[k][j] = green[k-gcnt][j];
                green[k-gcnt][j] = false;
            }
        }
    }

    cout << answer << "\n";

    int cnt = 0;
    for(int i=0; i<4; i++) {
        for(int j=0; j<6; j++) {
            if (blue[i][j]) cnt++;
            if (green[j][i]) cnt++;
        }
    }

    cout << cnt;

    return 0;
}