[Level 2] 삼각 달팽이 (Python)

코딩테스트 연습 - 삼각 달팽이

 

def solution(n):
    answer = []
    
    mtx = [[0]*(i+1) for i in range(n)]

    i = -1
    j = 0
    num = 0

    while True:
        for _ in range(n):
            num += 1
            i += 1
            mtx[i][j] = num 
        n -= 1

        if n == 0:
            break
        
        for _ in range(n):
            num += 1
            j += 1
            mtx[i][j] = num
        n -= 1

        if n == 0:
            break

        for _ in range(n):
            i -= 1
            j -= 1
            num += 1
            mtx[i][j] = num
        n -= 1

        if n == 0:
            break
        
    for i in mtx:
        for j in i:
            answer.append(j)
            
    return answer

 

▷ 가장 위에 위치한 꼭지점을 기준으로 테두리에 할당되는 숫자는 주어진 n에서 하나씩 줄어드는 규칙을 가지고 있다. 예를 들어, 왼쪽 위 첫 번째 테두리는 n, 다음 아래 테두리는 n - 1, 그 다음 오른쪽 위 테두리는 n - 2만큼 할당된다.

 

▷ 이때, 숫자가 할당될 때마다 1 씩 증가하여야 문제의 규칙을 만족하게 된다. 

 

▷ 구현하기 위해 i와 j는 할당되는 위치, num은 할당될 때의 숫자로 정의하고, 하나씩 줄어든 n이 0이 되는 경우 멈추도록 한다.

 

▶ for [Indicator] in [Array] 문에 N차원 배열을 넣게 되면, 배열의 원소인 N-1차원 배열을 순서대로 Indicator에 할당한다.