[프로그래머스 level2] 삼각달팽이

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

풀이

# n = 6인 경우
[1]
[2,15]
[3,16,14]
[4,17,21,13]
[5,18,19,20,12]
[6,7,8,9,10,11]

계속 3가지 패턴이 반복된다.
ans[x][y]라는 2차원 배열에 적는다고 가정하면,
1) 아래로 1 이동 x+=1 (1, 0)
2) 오른쪽으로 1 이동 y+=1 (0, 1)
3) 대각선 좌측 1 위로 1 이동 x-=1, y-=1 (-1, -1)
👉 %3 하면 되겠쥬

ex)
n = 6의 경우는
-6번(아래) 1,2,3,4,5,6 / i=0번째
-5번(오른쪽) 7,8,9,10,11 / i=1번째
-4번(대각선) 12,13,14,15 / i=2번째
-3번(아래) 16,17,18 / i=3번째
-2번(오른쪽) 19,20 / i=4번째
-1번(대각선) 21 / i=5번째
씩 찍는다.

방향을 dx, dy 리스트에 담아놓고 %연산에 따라 [아래,오른쪽,대각선] 방향으로 가도록 했다.
(x,y) = (-1,0)부터 출발하도록 했다.

이 패턴으로 해서 아래 코드로 작성.

코드

def solution(n):
    ans = [[0]*i for i in range(1, n+1)]
    num = 1
    dx = [1,0,-1]
    dy = [0,1,-1]
    x = -1
    y = 0
    for i in range(n):
        for _ in range(n-i):
            x += dx[i%3]
            y += dy[i%3]
            ans[x][y] = num
            num += 1
    return sum(ans, [])