[프로그래머스 / 파이썬] Lv.0 정수를 나선형으로 배치하기

https://school.programmers.co.kr/learn/courses/30/lessons/181832

 

문제

 

문제 설명

양의 정수 n이 매개변수로 주어집니다. n × n 배열에 1부터 n^2까지 정수를 인덱스 [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	[[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]]

 

 

소스 코드

def solution(n):
    if n == 1:
        return [[1]]

    answer = [[0 for j in range(n)] for i in range(n)]

    i = 0       # 행
    j = 0       # 열
    dir = 'r'   # 방향
    
    for num in range(n*n):
        answer[i][j] = num+1
        if dir == 'r':
            j += 1
            if j == n-1 or answer[i][j+1] != 0:
                dir = 'd'
        elif dir == 'd':
            i += 1
            if i == n-1 or answer[i+1][j] != 0:
                dir = 'l'
        elif dir == 'l':
            j -= 1
            if j == 0 or answer[i][j-1] != 0:
                dir = 'u'
        else:
            i -= 1
            if i == 0 or answer[i-1][j] != 0:
                dir = 'r'            
    
    
    return answer

 

풀이

• 1 x 1 배열인 경우에 [[1]]을 출력
• 행 번호 = i / 열 번호 = j / 방향 = dir (r:오른쪽 → d:아래쪽  → l:왼쪽 → u:위쪽)
• n x n 범위를 돌면서 숫자를 하나씩 채워주는 for문
  • 방향이 오른쪽인 경우 : 열 한 칸 이동 / 마지막 열이거나 칸이 채워져 있으면 방향 변경
  • 방향이 아래쪽인 경우 : 행 한 칸 이동 / 마지막 행이거나 칸이 채워져 있으면 방향 변경
  • 방향이 왼쪽인 경우 : 열 한 칸 앞으로 이동 / 첫 번째 열이거나 칸이 채워져 있으면 방향 변경
  • 방향이 위쪽인 경우 : 행 한 칸 위로 이동 / 첫 번째 행이거나 칸이 채워져 있으면 방향 변경

 

 

이거 레벨0 맞아요..? 그렇다면 내 실력이 마이너스인가보다....이런 저런 방식으로 해보다가 결국 검색해서 알게 되었는데 앞으로 더 많은 문제를 풀어봐야겠다는 생각이 든다...플러스 실력이 될 때까지 화이팅..!!