프로그래머스 - 삼각달팽이

python3

def solution(n):
    triangle = [[0 for _ in range(i+1)] for i in range(n)]
    x,y = -1, 0
    num = 1
    
    for i in range(n):
        for j in range(n-i):
            if i%3 == 0: #하
                x += 1
            elif i%3 == 1: #중
                y += 1
            else: # 상
                x -= 1
                y -= 1
            triangle[x][y] = num
            num += 1
        
    return sum(triangle, [])

그냥 삼각형이든 사각형이든 행이랑 열 있는거는 구현하고 x,y 이용해서 풀자,,