[백준] 15685 드래곤 커브

풀이

규칙을 찾아서 해결하는 문제이다. 규칙은 생각보다 잘 보이지 않는데, 드래곤 커브의 움직임을 화살표로 표현해서 생각해보자.

0세대 드래곤 커브는 → 으로 표현할 수 있다. 그리고 1세대 드래곤 커브는 → ↑ 이다. 마찬가지 방식으로 표현하면 2세대 드래곤 커브는 → ↑ ← ↑ 이 된다. 마지막으로 3세대 드래곤 커브는 → ↑ ← ↑ ← ↓ ← ↑ 로 표현된다.

규칙은 직전 세대의 드래곤 커브의 뒤에서부터 반시계 방향으로 90도 회전시켜주면 된다.

위의 그림과 같은 방식으로 N-1 세대를 알고 있다면 N 세대를 유추할 수 있다. 따라서 규칙에 맞게 드래곤 커브들을 모두 그려준 뒤, 1x1 크기의 정사각형의 개수를 세어주면 된다.

코드

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

int n;
bool map[101][101];
const int dx[] = { 1, 0, -1, 0 }, dy[] = { 0, -1, 0, 1 };

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

    cin >> n;
    for (int i = 0; i < n; i++) {
        int x, y, d, g;
        cin >> x >> y >> d >> g;
        vector<int> moves;
        moves.push_back(d);
        // 규칙 찾기
        for (int j = 0; j < g; j++) {
            int size = moves.size();
            for (int l = size - 1; l >= 0; l--) 
                moves.push_back((moves[l] + 1) % 4);
        }

        // 규칙에 맞게 드래곤 커브 그리기
        map[y][x] = true;
        for (int j = 0; j < moves.size(); j++) {
            int dir = moves[j];
            x += dx[dir];  y += dy[dir];
            map[y][x] = true;
        }
    }
    // 정사각형 개수 세기
    int ans = 0;
    for (int i = 0; i <= 99; i++) 
        for (int j = 0; j <= 99; j++) 
            if (map[i][j] && map[i + 1][j] && map[i][j + 1] && map[i + 1][j + 1]) ans++;
    cout << ans << '\n';
    return 0;
}