15685번: 드래곤 커브
첫째 줄에 드래곤 커브의 개수 N(1 ≤ N ≤ 20)이 주어진다. 둘째 줄부터 N개의 줄에는 드래곤 커브의 정보가 주어진다. 드래곤 커브의 정보는 네 정수 x, y, d, g로 이루어져 있다. x와 y는 드래곤 커
www.acmicpc.net
문제
드래곤 커브는 다음과 같은 세 가지 속성으로 이루어져 있으며, 이차원 좌표 평면 위에서 정의된다. 좌표 평면의 x축은 → 방향, y축은 ↓ 방향이다.
- 시작 점
- 시작 방향
- 세대
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인 정사각형의 네 꼭짓점이 모두 드래곤 커브의 일부인 것의 개수를 출력한다.
코드
dr = [0, -1, 0, 1]
dc = [1, 0, -1, 0]
map_data = [[False for _ in range(101)] for _ in range(101)]
for _ in range(int(input())):
c, r, d, g = map(int, input().split())
map_data[r][c] = True # 0세대 드래곤 커브 시작점 표시
dir_list = [d] # 방향 리스트
# 방향 리스트 만들기
for _ in range(g):
for i in range(len(dir_list)-1, -1, -1):
dir_list.append((dir_list[i]+1) % 4)
# 드래곤 커브
for dir in dir_list:
r += dr[dir]
c += dc[dir]
if r < 0 or r > 100 or c < 0 or c > 100:
continue
map_data[r][c] = True
# 정사각형 개수 체크
ans = 0
for i in range(100):
for j in range(100):
if map_data[i][j] and map_data[i][j+1] and map_data[i+1][j] and map_data[i+1][j+1]:
ans += 1
print(ans)
드래곤 커브는 모두 길이가 1인 선분들을 이어붙인 것이므로 방향 리스트만 있다면 커브를 쉽게 만들 수 있다.
문제에 주어졌던 3세대 드래곤 커브의 그림을 보면 n세대 드래곤 커브는 n-1세대 드래곤 커브를 현재 끝점에 90도 돌려서 붙여주면 된다는 것을 알 수 있다. 따라서 dir_list에 저장된 값들을 거꾸로 순회하며 dir_list에 추가해주면 되는데 이때 시계 방향으로 90도 돌려야하므로 1을 더한 값을 dir_list에 추가한다.
예시의 3세대 드래곤 커브를 구할 때 끝점에서 2세대 드래곤 커브를 거꾸로 순회하는 것을 그림으로 보면 다음과 같다. 동그라미의 숫자는 순회하는 순서이다.
0세대부터 g세대까지의 방향리스트들을 만들고 행(r)과 열(c) 값에 문제에서 주어진 방향의 순서 →(0) ↑(1) ←(2) ↓(3)에 따라 미리 정해놓은 dr, dc 값을 더해주면 된다.
'문제풀이 > 구현' 카테고리의 다른 글
[Python/파이썬] 백준 1244번 스위치 켜고 끄기 (0) | 2023.04.19 |
---|---|
[Python/파이썬] 백준 17140번 이차원 배열과 연산 (0) | 2023.03.06 |
[Python/파이썬] 백준 20056번 마법사 상어와 파이어볼 (0) | 2023.03.03 |
[Python/파이썬] 백준 17144번 미세먼지 안녕! (0) | 2023.03.03 |
[Python/파이썬] 백준 21610번 마법사 상어와 비바라기 (0) | 2023.03.02 |