[백준]15685: 드래곤 커브

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

15685번: 드래곤 커브

문제

드래곤 커브는 다음과 같은 세 가지 속성으로 이루어져 있으며, 이차원 좌표 평면 위에서 정의된다. 좌표 평면의 x축은 → 방향, y축은 ↓ 방향이다.

1. 시작 점
2. 시작 방향
3. 세대

0세대 드래곤 커브는 아래 그림과 같은 길이가 1인 선분이다. 아래 그림은 (0, 0)에서 시작하고, 시작 방향은 오른쪽인 0세대 드래곤 커브이다.

1세대 드래곤 커브는 0세대 드래곤 커브를 끝 점을 기준으로 시계 방향으로 90도 회전시킨 다음 0세대 드래곤 커브의 끝 점에 붙인 것이다. 끝 점이란 시작 점에서 선분을 타고 이동했을 때, 가장 먼 거리에 있는 점을 의미한다.

2세대 드래곤 커브도 1세대를 만든 방법을 이용해서 만들 수 있다. (파란색 선분은 새로 추가된 선분을 나타낸다)

3세대 드래곤 커브도 2세대 드래곤 커브를 이용해 만들 수 있다. 아래 그림은 3세대 드래곤 커브이다.

즉, K(K > 1)세대 드래곤 커브는 K-1세대 드래곤 커브를 끝 점을 기준으로 90도 시계 방향 회전 시킨 다음, 그것을 끝 점에 붙인 것이다.

크기가 100×100인 격자 위에 드래곤 커브가 N개 있다. 이때, 크기가 1×1인 정사각형의 네 꼭짓점이 모두 드래곤 커브의 일부인 정사각형의 개수를 구하는 프로그램을 작성하시오. 격자의 좌표는 (x, y)로 나타내며, 0 ≤ x ≤ 100, 0 ≤ y ≤ 100만 유효한 좌표이다.

입력

첫째 줄에 드래곤 커브의 개수 N(1 ≤ N ≤ 20)이 주어진다. 둘째 줄부터 N개의 줄에는 드래곤 커브의 정보가 주어진다. 드래곤 커브의 정보는 네 정수 x, y, d, g로 이루어져 있다. x와 y는 드래곤 커브의 시작 점, d는 시작 방향, g는 세대이다. (0 ≤ x, y ≤ 100, 0 ≤ d ≤ 3, 0 ≤ g ≤ 10)

입력으로 주어지는 드래곤 커브는 격자 밖으로 벗어나지 않는다. 드래곤 커브는 서로 겹칠 수 있다.

방향은 0, 1, 2, 3 중 하나이고, 다음을 의미한다.

• 0: x좌표가 증가하는 방향 (→)
• 1: y좌표가 감소하는 방향 (↑)
• 2: x좌표가 감소하는 방향 (←)
• 3: y좌표가 증가하는 방향 (↓)

출력

첫째 줄에 크기가 1×1인 정사각형의 네 꼭짓점이 모두 드래곤 커브의 일부인 것의 개수를 출력한다.

예제 입력 1 복사

3
3 3 0 1
4 2 1 3
4 2 2 1

예제 출력 1 복사

4

---

풀이

문제를 읽어본 후 이게 무슨 소리지?하고 한참을 쳐다봤던 것 같다.

"K(K > 1)세대 드래곤 커브는 K-1세대 드래곤 커브를 끝 점을 기준으로 90도 시계 방향 회전 시킨 다음, 그것을 끝 점에 붙인 것이다."

라는 구문의 이해가 안 되는 저같은 분을 위해 다음과 같은 그림이 필요할 것 같습니다.

1. 각 차수별 모든 방향을 추가해준다.

private static ArrayList<Integer> addDist(int d, int g) {
    ArrayList<Integer> dl = new ArrayList<>();
    dl.add(d);
    for(int i=0; i<g; i++) {
        for(int j=dl.size()-1; j>=0; j--) {
            dl.add((dl.get(j)+1)%4);
        }
    }

    return dl;
}

2. 추가된 방향을 모두 순회하며 map에 1로 체크해준다.

private static void setMap(int startX, int startY, ArrayList<Integer> dl) {
    int nx = startX;
    int ny = startY;
    map[nx][ny] = 1;
    for(int i=0; i<dl.size(); i++) {
        nx += dist[dl.get(i)][0];
        ny += dist[dl.get(i)][1];
        map[nx][ny] = 1;
    }
}

3. map을 완전탐색하면서 네모가 형성돼 있는 곳을 확인해준다.

for(int i=0; i<100; i++) {
    for(int j=0; j<100; j++) {
        if(map[i][j] == 1 && map[i+1][j] == 1 && map[i][j+1] == 1 && map[i+1][j+1] == 1) cnt++;
    }
}

전체 코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.StringTokenizer;

public class BJ_G4_15685_드래곤커브 {
    static int[][] dist = {{0, 1}, {-1, 0}, {0, -1}, {1, 0}};
    static int[][] map = new int[101][101];

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        StringTokenizer st;
        for(int i=0; i<N; i++) {
            st = new StringTokenizer(br.readLine());
            int y = Integer.parseInt(st.nextToken());
            int x = Integer.parseInt(st.nextToken());
            int d = Integer.parseInt(st.nextToken());
            int g = Integer.parseInt(st.nextToken());
            ArrayList<Integer> distList = addDist(d, g);
            setMap(x, y, distList);
        }

        int cnt = 0;
        for(int i=0; i<100; i++) {
            for(int j=0; j<100; j++) {
                if(map[i][j] == 1 && map[i+1][j] == 1 && map[i][j+1] == 1 && map[i+1][j+1] == 1) cnt++;
            }
        }
        System.out.println(cnt);
    }

    private static ArrayList<Integer> addDist(int d, int g) {
        ArrayList<Integer> dl = new ArrayList<>();
        dl.add(d);
        for(int i=0; i<g; i++) {
            for(int j=dl.size()-1; j>=0; j--) {
                dl.add((dl.get(j)+1)%4);
            }
        }

        return dl;
    }

    private static void setMap(int startX, int startY, ArrayList<Integer> dl) {
        int nx = startX;
        int ny = startY;
        map[nx][ny] = 1;
        for(int i=0; i<dl.size(); i++) {
            nx += dist[dl.get(i)][0];
            ny += dist[dl.get(i)][1];
            map[nx][ny] = 1;
        }
    }

    private static boolean isIn(int x, int y) {
        return 1<=x && x<=100 && 1<=y && y<=100;
    }
}