삼각 달팽이

def solution(n):
    arr = [[0 for i in range(n)] for i in range(n)]
    
    re = n*(n+1)//2
    
    ham(0,0,1,n,n,re,arr)
    
    answer = []
    for a in arr:
        for i in a:
            if i != 0:
                answer.append(i)
    return answer

def ham(r, c, cnt, r_num,c_num, renum,arr):
    for i in range(r, r_num):
        arr[i][c] = cnt
        cnt+=1
        if cnt == renum+1:
            return
    for j in range(c+1,c_num):
        arr[r_num-1][j] = cnt
        cnt+=1
        if cnt == renum+1:
            return
    for i in range(r_num-2,r,-1):
        arr[i][c_num-1] = cnt
        cnt+=1
        if cnt == renum+1:
            return
    ham(r+2,c+1,cnt,r_num-1,c_num-2, renum,arr)

정수 n이 매개변수로 주어집니다. 다음 그림과 같이 밑변의 길이와 높이가 n인 삼각형에서 맨 위 꼭짓점부터 반시계 방향으로 달팽이 채우기를 진행한 후, 첫 행부터 마지막 행까지 모두 순서대로 합친 새로운 배열을 return 하도록 solution 함수를 완성해주세요.

• 제한사항

n은 1 이상 1,000 이하입니다.

• 입출력 예

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