[월간코드] 삼각달팽이

문제 링크

프로그래머스

1. 문제 설명

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

---

2. 제한사항

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

---

3. 입출력 예

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]

---

4. 코드

def solution(n):
    answer = []
    # 2차원 배열 만들기
    tri = [[0 for col in range(n)] for row in range(n)]

    x = 0
    y = 0
    num = 1
    set_num = 1

    for i in range(n, 0, -1):

        if ((n-i) % 3 == 0):
            for num in range(1, i+1):
                tri[y][x] = set_num
                y += 1
                set_num += 1
            y -= 1
            x += 1

        if ((n-i) % 3 == 1):
            for num in range(1, i+1):
                tri[y][x] = set_num
                x += 1
                set_num += 1
            x -= 2
            y -= 1

        if ((n-i) % 3 == 2):
            for num in range(1, i+1):
                tri[y][x] = set_num
                set_num += 1
                x -= 1
                y -= 1
            y += 2
            x += 1

    for x in range(n):
        for y in range(n):
            if tri[x][y] > 0:
                answer.append(tri[x][y])

    return answer