[C++] [프로그래머스 2020 KAKAO] 기둥과 보 설치

#include <vector>
 
using namespace std;
 
struct coord {
    int pillar;
    int beam;
};
 
coord arr[101][101];
 
//기둥 세울 조건 바닥이거나 x=0, 보의 한쪽 끝부분(양쪽은 안됨), 다른 기둥 위
// 보 한쪽 끝부분이 기둥 위에 있거나, 양쪽 끝 부분이 다른 보와 연결
 
bool pillarCheck(int x, int y, int n) {
    if (y == 0) return true;
    if (arr[x][y - 1].pillar) return true;
    if (x > 0 && arr[x - 1][y].beam) return true;
    if (x < n && arr[x][y].beam) return true;
 
    return false;
}
 
bool beamCheck(int x, int y, int n) {
    if (arr[x][y - 1].pillar) return true;
    if (x < n && arr[x + 1][y - 1].pillar) return true;
    if (x > 0 && x < n && arr[x + 1][y].beam && arr[x - 1][y].beam) return true;
 
    return false;
}
 
vector<vector<int>> solution(int n, vector<vector<int>> build_frame) {
    vector<vector<int>> answer;
 
    for (int i = 0; i < build_frame.size(); i++) {
        int x = build_frame[i][0];
        int y = build_frame[i][1];
        int a = build_frame[i][2];
        int b = build_frame[i][3];
 
        if (b == 1 && a == 0) {//기둥 설치
            if (pillarCheck(x, y, n)) arr[x][y].pillar = 1;
        }
        else if (b == 1 && a == 1) {// 보 설치
            if (beamCheck(x, y, n)) arr[x][y].beam = 1;
        }
        else if (b == 0 && a == 0) {// 기둥 삭제
            bool check = true;
            arr[x][y].pillar = 0;
 
            if (y < n && arr[x][y + 1].pillar && !pillarCheck(x, y + 1, n)) check = false;
            else if (y < n && arr[x][y + 1].beam && !beamCheck(x, y + 1, n)) check = false;
            else if (x > 0 && arr[x - 1][y + 1].beam && y < n && !beamCheck(x - 1, y + 1, n)) check = false;
 
            if (!check) arr[x][y].pillar = 1;
 
        }
        else if (b == 0 && a == 1) {// 보 삭제
            bool check = true;
            arr[x][y].beam = 0;
 
            if (arr[x][y].pillar && !pillarCheck(x, y, n)) check = false;
            else if (x < n && arr[x + 1][y].pillar && !pillarCheck(x + 1, y, n)) check = false;
            else if (x > 0 && arr[x - 1][y].beam && !beamCheck(x - 1, y, n)) check = false;
            else if (x < n && arr[x + 1][y].beam && !beamCheck(x + 1, y, n)) check = false;
 
            if (!check) arr[x][y].beam = 1;
        }
 
    }
 
    for (int i = 0; i <= n; i++) {
        for (int j = 0; j <= n; j++) {
            if (arr[i][j].pillar) {
                answer.push_back({ i, j, 0 });
            }
            if (arr[i][j].beam) {
                answer.push_back({ i, j, 1 });
            }
        }
    }
 
 
    return answer;
}