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

문제 설명

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

---

제한사항

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

---

입출력 예

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]

---

내 코드

def solution(n):
    answer = []
    arr = [[0]*n for _ in range(n)]
    x,y = -1,0
    num = 1 
    
    for i in range(n):
        for _ in range(i,n):
            if i % 3 == 0: # 아래
                x += 1
            elif i % 3 == 1: # 오른쪽
                y += 1
            else: # 대각선
                x -= 1
                y -= 1
            arr[x][y] = num 
            num += 1
    
    for a in arr:
        for j in a:
            if j != 0:
                answer.append(j)
                
    return answer

아이디어

1. n*n 2차원 배열로 만들어서 값을 할당하기
   이런 류의 문제들은 대개 2차원 배열로 만들어 놓고 생각해보자. 왼쪽 정렬로 숫자들을 밀어 넣고 그 다음 빈 곳은 0으로 값을 채운 뒤 규칙을 찾아본다.

2. 삼각 달팽이의 값이 바뀌는 지점을 체크할 것
   문제에서 주어진 예시 그림에서 꺾이는 지점을 체크해야 한다.

   4*4 2차원 배열을 예로 들면, (0,0) 에서 x좌표의 0 지점에서 아래로 내려갔다가, (3,0)에서는 오른쪽으로 이동 후 그리고 (3,3) 지점에 도달 했을때는 대각선의 윗방향으로 올라가는 규칙성을 볼 수 있다.

   이에 맞춰서 삼각(3변)의 값의 나머지로 분기문을 만들어 위치값을 조정하는 로직을 짜면 된다.