👩‍💻 코테 공부/코테 공부

[코테 - py] 백준 15685 드래곤 커브

수댕ʕت̫͡ʔ 2024. 8. 12. 17:29

문제

드래곤 커브는 다음과 같은 세 가지 속성으로 이루어져 있으며, 이차원 좌표 평면 위에서 정의된다. 좌표 평면의 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

 

 

내가 푼 풀이

"""
규칙을 찾자!
-> / 위
위 / <-
<- / 아래
아래 / ->
"""
num = int(input())

graph = [[0 for _ in range(101)] for _ in range(101)]

# 규칙 순!
directions = [(1, 0), (0, -1), (-1, 0), (0, 1)]

for _ in range(num):
    x, y, d, g = map(int, input().split(' '))
    graph[x][y] = 1

    move = [d]

    # 규칙에 의해서 순서를 담자!
    for _ in range(g):
        temp = []
        for i in range(len(move)):
            temp.append((move[-i-1]+1) % 4)
        move.extend(temp)

    # 규칙대로 움직여주자!
    for i in move:
        next_r = x + directions[i][0]
        next_c = y + directions[i][1]
        graph[next_r][next_c] = 1
        x, y = next_r, next_c

answer = 0
for i in range(100):
    for j in range(100):
        if graph[i][j] == 1 and graph[i+1][j] == 1 and graph[i][j+1] == 1 and graph[i+1][j+1] == 1:
            answer += 1

print(answer)

 

규칙을 찾는것이 키포인트 였다!

 

엄청 고민하다가 규칙을 발견했다. 하지만 그 뒤를 어떻게 해결해야할지 고민하다가 풀이를 보고 힌트를 얻었다.

규칙은 다음과 같다.

기준  오른쪽 왼쪽 아래
추가 왼쪽 아래 오른쪽

 

이렇게 방향을 기준으로 방향을 추가해주고, 추가한 다음 선을 그린다.

 

즉, 예를들어 왼쪽 방향으로 움직였다면 아래로 움직여줘야함.

 

이 규칙을 살려서 추가해주면 선의 방향을 저장해줄수있고, 그걸 그려주면된다.

 

구현은 재밌지만, 아이디어를 생각하기에는 어려운것같다..!