[Python] 삼각 달팽이

문제 설명

 

정수 n이 매개변수로 주어집니다. 다음 그림과 같이 밑변의 길이와 높이가 n인 삼각형에서 맨 위 꼭짓점부터 반시계 방향으로 달팽이 채우기를 진행한 후, 첫 행부터 마지막 행까지 모두 순서대로 합친 새로운 배열을 return 하도록 solution 함수를 완성해주세요.

 

 

입출력 예

 

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차원 리스트로 생각

2) 꺾기 전 한줄의 개수가 n개임 -> i을 0부터 n까지 증가시키면서 각 좌표의 값 채우기

   *** num 값 증가에 따른 좌표값 이동

def solution(n):
    arr = [[0 for j in range(n)] for i in range(n)]
    x, y = -1, 0 # *** 좌표 초기화
    # 처음 시작은 아래로 내려가서 x = -1
    num = 1
    
    for i in range(n): # 꺾기 전 한 줄을 기준으로 
        for j in range(i, n):
            if i % 3 == 0:
                x += 1
            elif i % 3 == 1:
                y += 1
            else:
                x -= 1
                y -= 1
            arr[x][y] = num
            num +=1
    # n이 4일때 
    # arr은 [[1, 0, 0, 0], [2, 9, 0, 0], [3, 10, 8, 0], [4, 5, 6, 7]]
    # 1차원 list로 변환
    answer = []
    for i in arr:
        for j in i:
            if j:
                answer.append(j)
    return answer

 

 

2차원을 1차원으로 바꿀 때 sum(list, []) 사용

단, arr에 item 중 0이 있으면 안됨

def solution(n):
    arr = [[0 for j in range(1, i+1)] for i in range(1, n+1)]
    x, y = -1, 0 # *** 좌표 초기화
    # 처음 시작은 아래로 내려가서 x = -1
    num = 1
    
    for i in range(n): # 꺾기 전 한 줄을 기준으로 
        for j in range(i, n):
            if i % 3 == 0:
                x += 1
            elif i % 3 == 1:
                y += 1
            else:
                x -= 1
                y -= 1
            arr[x][y] = num
            num +=1
    # n이 4일때 
    # arr은 [[1], [2, 9], [3, 10, 8], [4, 5, 6, 7]]
    print(arr)
    # 1차원 list로 변환
    return sum(arr, [])
solution(4)