삼각 달팽이

나의 코드

def solution(n):

    answer = []

    tree_list = []

    for _ in range(n):

        tree_list.append([0]*n)

    #print(tree_list)

    x,y = -1,0

    number = 1

    for i in range(n):

        for j in range(i,n):

            if i%3 == 0:

                x+=1

                tree_list[x][y] = number

                number += 1

            elif i%3 == 1:

                y+=1

                tree_list[x][y] = number

                number+=1

            else:

                x-=1

                y-=1

                tree_list[x][y] = number

                number+=1

    #print(tree_list)

    for i in tree_list:

        for j in i:

            if j!=0:

                answer.append(j)

    #print(answer)

    return answer

solution(4)

점점 어려워진다... ㅠ.ㅠ.ㅠㅠ