[프로그래머스] LV.2 삼각 달팽이

[프로그래머스] LV.2 삼각 달팽이

 

1. 문제 : Link

삼각형으로 돌아가면서 숫자를 채워넣는 삼각달팽이 완성하는 문제

 

2. 풀이

먼저 삼각 달팽이의 껍데기를 만들어준다.

반복문을 통해 회전하면서 값을 대입해준다.

 

3. 코드

def solution(n):
    res = [[0] * n for _ in range(n)]
    answer = []
    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
            
			#저장  
            res[x][y] = num
            num += 1
            
    for i in res:
        for j in i:
            if j != 0:
                answer.append(j)
    return answer

* 참고) 스킬트리 for-esle문