[baekjoon] 백준 1913번(파이썬): 달팽이

문제

1913번: 달팽이

알고리즘

- 반복문을 통해 달팽이를 움직인다.

- 우/하/좌/상을 현재 길이 만큼 움직인다.

- 움직인 후 달팽이의 위치를 왼쪽 모서리로 움직여준다.

코드

import sys

n = int(sys.stdin.readline())
m = int(sys.stdin.readline())
graph = [[0 for _ in range(n)] for _ in range(n)]

# 우/하/좌/상
dx = [0, 1, 0, -1]
dy = [1, 0, -1, 0]
x = n // 2 # 처음 x 좌표
y = n // 2 # 처음 y 좌표
num = 1
l = 0
graph[x][y] = 1
answer = []

# 반복문을 통해 달팽이를 움직인다.
while True:
    # 우/하/좌/상 이동
    for i in range(4):
        # 길이만큼 이동
        for _ in range(l):
            x += dx[i]
            y += dy[i]
            num += 1
            graph[x][y] = num

    # 달팽이가 움직였을 때 좌표가 0,0이라면 반복을 멈춤
    if x == y == 0:
        break

    # 왼쪽 모서리로 이동
    x -= 1
    y -= 1
    l += 2  # 달패이가 움직이는 직선 길이

for j in range(len(graph)):
    print(*graph[j])
    if m in graph[j]:
        mx = j + 1
        my = graph[j].index(m) + 1

print(mx, my)

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법(Python)