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

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

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

 

 

슬슬 이게 레벨 2라고?!!?!? 하는 문제들이 나온다...🤦‍♀️ 벌써 레벨 3이 무섭다...... 

 

좌표를 이동해서 풀어야겠다고 생각은 했으나 n*n 배열이 아니라

1, 2, 3, 4, 5.... 이런 식으로 이차원 배열을 만들려고 했음 ( [ [0], [0,0], [0,0,0], [0,0,0,0] .... ] )

그러다 너무 어려워서 다른 사람들은 어떻게 풀었나 선조들의 지혜를 빌렸음..

 

그래서 위에 말했듯이 n*n 배열을 만들어서 숫자를 넣은 후, 0을 지워나가면 되는 방법을 깨달았다.

 

밑에는 문제 풀고 바로 까먹을까봐 정리한거!

 

4*4 만 보면 헷갈릴까봐 더 큰 배열도 준비했습니다. 미래의 날 위해서...

 

이정도 친절함이면 미래의 내가 알아보겠지...? 😢

 

잘 모르겠으면 코드 한번 보고 그림 한번 보고 하면서 이해해봅시다!

 

 

🙋‍♀️ 작성한 코드

 

def solution(n):
    answer = []
    
    tmp_list = [[0 for _ in range(n)] for _ in range(n)]
    
    x = 0 #가로
    y = -1 #세로
    
    num = 1
    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: #왼쪽 위 대각선 이동
                x -= 1
                y -= 1
            tmp_list[y][x] = num #좌표에 숫자를 입력
            num += 1
    
    for tmp in tmp_list:
        for t in tmp:
            if t != 0:
                answer.append(t)
    
    return answer