프로그래머스 LV0 정수를 나선형으로 배치하기

문제 설명

양의 정수 n이 매개변수로 주어집니다. n × n 배열에 1부터 n2 까지 정수를 인덱스 [0][0]부터 시계방향 나선형으로 배치한 이차원 배열을 return 하는 solution 함수를 작성해 주세요.

제한 사항

1 ≤ n ≤ 30

입출력 예

n	result
4	[[1, 2, 3, 4], [12, 13, 14, 5], [11, 16, 15, 6], [10, 9, 8, 7]]
5	5 [[1, 2, 3, 4, 5], [16, 17, 18, 19, 6], [15, 24, 25, 20, 7], [14, 23, 22, 21, 8], [13, 12, 11, 10, 9]]

입출력 예 설명

• 입출력 예 #1
  • 예제 1번의 n의 값은 4로 4 × 4 배열에 다음과 같이 1부터 16까지 숫자를 채울 수 있습니다.
  • 따라서 [[1, 2, 3, 4], [12, 13, 14, 5], [11, 16, 15, 6], [10, 9, 8, 7]]를 return 합니다.

• 입출력 예 #2
  • 예제 2번의 n의 값은 5로 5 × 5 배열에 다음과 같이 1부터 25까지 숫자를 채울 수 있습니다.
  • 따라서 [[1, 2, 3, 4, 5], [16, 17, 18, 19, 6], [15, 24, 25, 20, 7], [14, 23, 22, 21, 8], [13, 12, 11, 10, 9]]를 return 합니다.

풀이

• 나선 방향으로 (RDLU) 회전한다.
• 처음 n만큼 가고, 그 다음엔 n-1만큼 두번 간다.
  • 예#1 : 4, 3, 3, 2, 2, 1, 1
  • 예#2 : 5, 4, 4, 3, 3, 2, 2, 1, 1
• 그래서 배열 2개 만들어서 구현.

def solution(n):
    snake = [n]
    d = ((0,1),(1,0),(0,-1),(-1,0))
    answer = [[0 for i in range(n)] for j in range(n)]
    count,x,y = 0,-1,0
    
    while n > 1:
        n -= 1
        snake.append(n)
        snake.append(n)
    
    for i in range(len(snake)):
        for j in range(snake[i]):
            y += d[i%4][0]
            x += d[i%4][1]
            count += 1
            answer[y][x] = count
        
        
    return answer