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

출처::  월간 코드 챌린지 시즌1  programmers.co.kr/learn/courses/30/lessons/68645

분류:: 구현, bfs

 

1. 문제 이해 및 해결과정

ex) n=4 1 2 9 3 10 8 4  5  6  7 - 방향 규칙 : 아래, 오른쪽, 왼쪽 대각선 반복 - 한 방향에 cnt만큼 채워짐 , cnt= 4,3,2,1

 

2. 풀이방법

  1. [ python ] 

def solution(n):
    count = [x for x in range(n, 0, -1)]  # n=4라면, 4,3,2,1개 입력으로 방향이 바뀜
    arr = [[0] * n for _ in range(n)]  # 배열
    dh = [1, 0, -1]  # 아래,오른쪽, 왼쪽대각선
    dw = [0, 1, -1]
    answer=[] #답

    h, w = 0, 0
    num = 1  # 입력되는 수
    dir = 0
    arr[h][w]=num
    num+=1
    while count:
        curCnt = count.pop(0) #현재방향의 개수
        for i in range(curCnt):
            nh = h + dh[dir]
            nw = w + dw[dir]
            if 0 <= nh < n and 0 <= nw < n and arr[nh][nw]==0: #아직 채워진 숫자가없으면
                arr[nh][nw] = num
                num += 1
                h,w=nh,nw #새로운좌표로 이동 
        dir = (dir + 1) % 3  # 방향변경

    for i in range(n):
        for j in range(n):
            if arr[i][j]!=0:
                answer.append(arr[i][j])
    # print(answer)
    return answer

 

3. 오답원인

 

4. 알게된 점