프로그래머스 코딩테스트(lv2) - 삼각달팽이

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]

[2, 9]

[3, 10, 8]

[4, 5, 6, 7]

 

• 빈리스트 만들기
• 3가지 케이스에 따라 움직임(좌표) 설정
• 만들어진 리스트를 하나의 리스트로 합치기

def solution(n):
    answer = []
    
    #빈 리스트 만들기
    tempArr = []
    for i in range(n):
        tempArr.append([0]*(i+1))
    
    #default 좌표 및 채울 숫자 설정
    x,y = 0, -1
    number = 1
    
    #이중 for문을 이용하여 3가지 케이스에 맞춰 이동
    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
            tempArr[y][x] = number
            number+=1
 
    #만들어진 2차원 리스트를 1차원 리스트로 변환
    for li in tempArr:
        for i in li:
            if i != 0:
                answer.append(i)
    return answer

끝.