[Programmers] 삼각 달팽이 - 월간 코드 챌린지1 (python)

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

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

def solution(n):

    li = [[0] * n for _ in range(n)]

    ret = []

    x, y = -1, 0

    num = 1

    for i in range(n):

        for j in range(i, n):

            if i % 3 == 0:

                x += 1

            elif i % 3 == 1:

                y += 1

            elif i % 3 == 2:

                x -= 1

                y -= 1

            li[x][y] = num

            num += 1

    for i in li:

        for j in i:

            if j != 0:

                ret.append(j)

    return ret