[PG] 삼각 달팽이

문제

me

import math
def solution(n):
    res = [[0] * n for _ in range(n)]
    row,col = -1,0
    curNum=1
    for i in range(n):
        for j in range(i,n):
            if i%3==0:
                row+=1
            elif i%3==1:
                col+=1
            else:
                row-=1
                col-=1
            res[row][col]=curNum
            curNum+=1
    arr=[]
    for i in range(n):
        for j in range(n):
            if res[i][j]!=0:
                arr.append(res[i][j])
    return arr

• 삼각형이고 규칙을 찾아보면 꺾는 위치가 하나씩 줄어든 다음이다.
• 처음에는 숫자로 일일히 하려고 했는데 좀 더 간편한 방법이 있다..
• 삼각형을 아예 2차원 배열로 풀어놓고 생각해도 좋은 것 같다.