[SWEA] 1954 달팽이 숫자 (파이썬/python)

🎈문제

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&problemLevel=3&problemLevel=4&contestProbId=AV5PobmqAPoDFAUq&categoryId=AV5PobmqAPoDFAUq&categoryType=CODE&problemTitle=&orderBy=SUBMIT_COUNT&selectCodeLang=JAVA&select-1=4&pageSize=10&pageIndex=1

 

🎁알고리즘 및 접근

구현

1. (0,-1)에서 시작해서 오른쪽으로 이동하며 리스트 바깥으로 나가면 방향을 아래로 바꾼다

2. 아래쪽, 왼쪽, 위쪽 모두 차례로 반복한다.

 

💻코드

# sw 1954 달팽이 숫자

def Snail(t,n):
    lst = [[0 for _ in range(n)]for _ in range(n)]
    dx = [0,1,0,-1] # 우, 하, 좌, 상
    dy = [1,0,-1,0]
    x,y = 0,-1   # 오른쪽으로 +1 하면서 시작하므로 초기좌표 0,-1로 설정
    direction = 0   # 방향을 설정한 순서대로 바꿀 때 사용할 변수
    for i in range(1,n*n+1):
        x += dx[direction]
        y += dy[direction]
        if x < 0 or x > (n-1) or y < 0 or y > (n-1) or lst[x][y] != 0:    # lst 밖으로 나가면
            x -= dx[direction]  # 다시 빼주고
            y -= dy[direction]
            direction = (direction + 1) % 4 # 방향 바꿔주고
            x += dx[direction]  # 바꾼 방향으로 다시 더해주고
            y += dy[direction]
        lst[x][y] = i
    
    print(f"#{t}")
    for i in range(n):
        print(*lst[i])

T = int(input())    # 테스트 케이스 개수

for t in range(1,T+1):
    n = int(input())    # 달팽이 크기
    Snail(t,n)

🎄코드 설명

주석과 위의 설명으로 대체한다.