[프로그래머스] 삼각 달팽이(Python)

https://programmers.co.kr/learn/courses/30/lessons/68645

코딩테스트 연습 - 삼각 달팽이

문제 설명

정수 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]

풀이 설명

1. 0으로 이루어진 밑변과 높이가 n인 삼각형 모양의 2차원 배열(arr)을 만든다.

2. 열심히 다양한 크기의 삼각형 그림 그려가며 규칙을 찾은 결과

3. 숫자를 3으로 나눈 나머지에 따라 그 숫자가 들어갈 자리(x,y인덱스)가 설정된다

   (직접 해 보면 알겠지만 글로 설명하기가 너무 어렵다..설명에 재능 전혀 없는 편..코드를 보면 이해될것임!)

4. arr을 다 채우고 각 원소배열을 합친 새로운 배열 리턴

Python3 코드

def solution(n):
    answer = []
    arr = []
    x, y = 0, -1
    count = 1
    
    for i in range(1, n+1):
        arr.append([0]*i)
        
    for i in range(n):
        for j in range(i, n):
            if i % 3 == 0:
                y += 1
            elif i % 3 == 1:
                x += 1
            elif i % 3 == 2:
                y -= 1
                x -= 1
            arr[y][x] = count
            count += 1
            
    for a in arr:
        answer.extend(a)
    return answer