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

목차

• 문제 설명
• 제한 사항
• 입출력 예
• 풀이과정
• 정답

문제 설명

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

풀이과정

• down (↓), right (→), up (↖) 방향이 반복됨
• 순서대로 6, 5, 4, 3, … 순서로 채우는 칸의 수가 줄어듦

정답

def solution(n):
    answer = []
    arr = [[0 for _ in range(i+1)] for i in range(n)]
    total_num = sum([i+1 for i in range(n)])
    
    d, i, j, num = 0, 0, 0, 1
    
    di = [1, 0, -1]
    dj = [0, 1, -1]
    
    move_i = [-1, -1, 2]
    move_j = [1, -2, 1]
    
    while num <= total_num:
        for _ in range(n):
            arr[i][j] = num
            i += di[d]
            j += dj[d]
            num += 1
        
        i += move_i[d] # 시작 위치로 옮겨주기
        j += move_j[d] # 시작 위치로 옮겨주기
        n -= 1 # 채우는 칸의 수가 줄어듦
        d = (d+1)%3 # 반복되는 방향
    
    for tmp in arr:
        answer += tmp
    
    return answer