[C++] 테트로미노 (백준 14500번)

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

 

#include <iostream>
#include <vector>
using namespace std;

int dx[][3] = {
    {1, 1, 1}, {0, 0, 0},
    {1, 0, -1},
    {0, 0, 1}, {1, 0, 0}, {0, 0, 1}, {1, 0, 0}, 
    {0, 1, 1}, {1, 1, 0}, {0, 1, 1}, {1, 1, 0},
    {0, 1, 0}, {0, -1, 0},
    {-1, 0, -1}, {-1, 0, -1},
    {-1, 1, 1}, {-1, 1, 1}, {1, 0, 0}, {-1, 0, 0}
};

int dy[][3] = {
    {0, 0, 0}, {1, 1, 1},
    {0, 1, 0},
    {-1, -1, 0}, {0, -1, -1}, {1, 1, 0}, {0, 1, 1},
    {1, 0, 0}, {0, 0, 1}, {-1, 0, 0}, {0, 0, -1},
    {-1, 0, -1}, {-1, 0, -1},
    {0, 1, 0}, {0, -1, 0},
    {1, 0, 0}, {-1, 0, 0}, {-1, 1, 1}, {-1, 1, 1}
};

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

    int n, m;
    cin >> n >> m;
    vector<vector<int>> g(n, vector<int>(m));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> g[i][j];
        }
    }
    
    int maxt = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            for (int k = 0; k < 19; k++) {
                int sum = g[i][j];
                int curi = i;
                int curj = j;
                bool psb = true;
                for (int l = 0; l < 3; l++) {
                    curi += dx[k][l];
                    curj += dy[k][l];
                    if (curi < 0 || curi >= n || curj < 0 || curj >= m) {
                        psb = false;
                        break;
                    }
                    sum += g[curi][curj];
                }
                if (psb) {
                    if (sum > maxt) maxt = sum;
                }
            }
        }
    }

    cout << maxt << '\n';
}

아이디어 자체는 전혀 어려울 것이 없는 단순 구현 문제였다. 하지만 수많은 '틀렸습니다'가 날 괴롭힌... 5가지 모양을 대칭 및 회전시키면 총 19개의 도형이 존재한다는 것을 알 수 있다. 그래서 나는 맨 위에 dx, dy로 19개를 정의해두고 반복문으로 해결했다. 아까 말했듯 아이디어 자체는 어렵지 않으나 도형에 맞추어 좌표를 생각하는게 은근.. 아니 많이.. 헷갈려서 넘 힘들었다. 암튼 겨우 성공. (AC)

 

후우................^^