[프로그래머스] 삼각 달팽이 / 파이썬 / Python / 월간 코드 챌린지 시즌1

💡solutions )

💬 피라미드 배열 만들고, 삼각형을 기준으로 if문 처리 (각 선분에 해당하는 3가지 조건으로 처리)

💬 해당 문제를 풀면서 list index out of range가 많이 나서 고생을 좀 했다..!!!!😂

💬 인덱스 에러가 나지 않게 while문 안에서 break 조건을 적절히 넣어줘야 한다

    ->j, k 범위를 체크하거나 arr[j][k] 값이 있는지 확인하는 조건을 넣었음

💬 최근 본 코테에서 2차원 배열 리스트를 활용하는 문제가 나왔는데 상당히 어렵게 느껴졌다. 해당 유형의 문제를 더 연습해야겠다!

 

🎫code )

def solution(n):
    arr = [[0] * _ for _ in range(1, n + 1)]
    num = 1
    j = 0
    k = 0

    for i in range(1, n+1):
        # 왼쪽 아래로 내려가는 대각선
        if i % 3 == 1:
            while True:
                arr[j][k] = num
                num += 1
                j += 1
                if j >= n or arr[j][k]:
                    j -= 1
                    k += 1
                    break
        # 왼쪽에서 오른쪽 향하는 일직선
        elif i % 3 == 2:
            while True:
                arr[j][k] = num
                num += 1
                k += 1
                if k >= n or arr[j][k]:
                    k -= 2
                    j -= 1
                    break
        # 왼쪽 위로 향하는 대각선
        elif i % 3 == 0:
            while True:
                arr[j][k] = num
                num += 1
                j -= 1
                k -= 1
                if arr[j][k]:
                    j += 2
                    k += 1
                    break

    return sum(arr, [])

 

📌 description )

문제출처 : programmers.co.kr/learn/courses/30/lessons/68645?language=python3

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

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