[프로그래머스] 삼각 달팽이 (level2, python)

문제 설명

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

 

 

---

제한사항

• n은 1 이상 1,000 이하입니다.

 

---

입출력 예

---

입출력 예 설명

 

입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• 문제 예시와 같습니다.

입출력 예 #3

• 문제 예시와 같습니다.

---

풀이 코드

def solution(n):
    size = n * (n+1) // 2
    answer = [0 for _ in range(size)]
    checker = [0, 0]
    
    idx = 0
    step = 1
    temp = n
    
    for i in range(1, size + 1):
        answer[idx] = i
        checker[1] += 1
        
        if checker[1] == temp:
            # 해당 방향으로 다 이동했으면,
            temp -= 1
            checker[1] = 0
            checker[0] = (checker[0] + 1) % 3
            
        direction = checker[0]
        if direction == 0:
            # down
            idx += step
            step += 1
        elif direction == 1:
            # right
            idx += 1
        elif direction == 2:
            # up
            idx -= step
            step -= 1
        
    return answer

 

별 다른 알고리즘 없이, 규칙에 의한 구현으로 풀이했다.

n = k일때, 피라미드의 블럭의 전에 수는 k * (k+1) // 2에 해당한다.

그 후, 규칙에 맞게 피라미드 내의 수를 채워 나간다.

 

삼각 달팽이의 이동은, 아래 → 오른쪽 → 위 → 아래 방향으로 이동한다.

아래로 갈 때마다 index의 step은 1씩 늘어난다.

n=4일 때의 경우, 1(1) → 2(2) → 3(4) →4(7)이므로, step만 보면 1 → 2 → 3 (여기까지 값 매김) → 4 (step만 설정하고 값 안 매김)으로 변환된다.

 

오른쪽으로 이동하면 index는 1씩 늘어나므로, step을 따로 설정할 필요 없다.

 

위로 이동하면, index는 이전 아래 이동에서의 step 값만큼 차감하며 이동한다. 그리고 위로 이동할 때마다 step은 1씩 줄어든다.

7 (10) → 8 (6) → 9 (3)은, 4 → 3 (여기까지 값 매김) → 2 (step 설정만 하고 값 안 매김)으로 볼 수 있다.

이 규칙을 이용해 구현하면 풀 수 있다.

 

level2 이지만, 은근히 생각이 필요한 구현문제라 정답률이 다소 낮은 모습을 볼 수 있었다.