백준 알고리즘 15685번 드래곤 커브 풀이 with Python

문제 링크: https://www.acmicpc.net/problem/15685

15685번: 드래곤 커브

문제는 아래와 같습니다.

백준 15685번 드래곤 커브

이 문제의 핵심은 드래곤 커브를 받았을 때 그 드래곤 커브를 알맞게 구현하는 것과 드래건 커브의 포인트들 중에서 사각형을 찾는 것이 핵심입니다. 아래는 제가 해결한 코드입니다. 

# x,y와 방향 d, 세대 g 그리고 드래곤 커브의 좌표값을 저장할 딕셔너리 dict1
def dragon(x, y, d, g, dict1):
    dx = [1, 0, -1, 0]
    dy = [0, -1, 0, 1]
    dict1[(x, y)] = 1
    list1 = [d]
    x += dx[d]
    y += dy[d]
    dict1[(x, y)] = 1
    for _ in range(g):
        sub = []
        for i in range(len(list1)-1, -1, -1):
            x += dx[(list1[i] + 1) % 4]
            y += dy[(list1[i] + 1) % 4]
            dict1[(x, y)] = 1
            sub.append((list1[i] + 1) % 4)
        list1 += sub


n = int(input())
answer = 0
dic1 = {}
# n개의 드래곤 커브를 dict1에 좌표값에 저장한다.
for _ in range(n):
    x, y, d, g = map(int, input().split())
    dragon(x, y, d, g, dic1)
# 좌표값들 중에 나는 왼쪽 상단을 기준으로 3개의 점이 있으면 사각형이 된다고 하였다.
for i in list(dic1.keys()):
    count = 0
    nx = i[0]
    ny = i[1]
    if dic1.get((i[0] + 1, i[1])) == 1:
        count += 1
    if dic1.get((i[0], i[1] + 1)) == 1:
        count += 1
    if dic1.get((i[0] + 1, i[1] + 1)) == 1:
        count += 1
    if count == 3:
        answer += 1
print(answer)