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

문제

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

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

 

코드

def solution(n):
    answer = [[0 for j in range(i+1)] for i in range(n)]
    result = []
    x,y = -1, 0
    num = 1
    for i in range(n):
        for j in range(i, n):
            #down
            if i % 3 == 0:
                x += 1
            #right
            elif i % 3 == 1:
                y += 1
            #up
            elif i % 3 == 2:
                x -= 1
                y -= 1
            answer[x][y] = num
            num += 1
    for i in range(n):
        for j in range(i+1):
            result.append(answer[i][j])
    return result

이중 리스트를 이용해

1

2 9

3 10 8

4 5 6 7

이런식으로 만들어주면 되는데 일단 answer을 [[0], [0,0], [0,0,0], [0,0,0,0]] 0을 채워 넣어준다.

 

값이 증가하는 방향을 보면 맨처음에는 아래로 갔다가 그다음에는 오른쪽 다음에는 대각선 왼쪽으로 가고 다음에는 앞에 순서를 반복한다. 그러므로 3가지 방향이 존재하여

 

i%3 == 0일 경우 아래로 이동할 수 있게 x에 +1을 해주고 (맨 처음은 무조건 아래로 이동하므로 x를 -1로 선언해준다.)

i%3 == 1일 경우 오른쪽으로 이동할 수 있게 y에 +1을 해준다.

i%3 == 2일 경우 대각선으로 이동할 수 있게 x와 y에 -1을 해준다.

 

방향이 정해지면 두번째 for문에서 처음에는 n만큼 for문을 돌아주고 그다음에는 모서리가 겹치기 때문에 i가 증가하는걸 이용해 for문을 조금씩 덜 돌게 된다.