프로그래머스 - 삼각 달팽이Swift(구현)

삼각 달팽이

문제 설명

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

---

제한사항

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

---

입출력 예nresult

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]

---

풀이 및 소스코드

2차원 배열로 생각해본다면 아래와 같이 표현할 수 있다

n = 5

[1,  0,  0,  0,  0]

[2, 12,  0, 0, 0]

[3, 13, 11, 0, 0]

[4, 14, 15, 10, 0]

[5,   6,   7,   8,   9]

방향 전환은 n회 일어나는데 방향 전환의 종류는 3이다. % 3 나머지 연산을 통한다면 n이 몇이 들어오던 0, 1, 2가 나오므로

3종류의 반복이 일어날 수 있다. 또한 한 번에 적히는 수의 개수도 아래 그림과 같다

n개부터 시작하여 n-1개를 반복하는걸 알 수 있다.

 

또한 방향전환에선 하강시 x좌표의 증가, 우측: y좌표의 증가, 상승:  x -= 1, y -= 1

의 좌표 변경을 볼 수 있다.

 

조건과 규칙을 모두 찾았으니 코드로 구현을 한다면

import Foundation

func solution(_ n:Int) -> [Int] {
    var pingping = [[Int]](repeating: [Int](repeating: 0, count: n), count: n)
    var curnum = 1
    var x = -1, y = 0 //시작시 x좌표가 증가하므로 -1로 초기화 시킨다.
    
    var result = [Int]()
    //첫 시작은 하강, 방향 전환은 n회 일어난다.
    for i in 0..<n{
        for _ in i..<n{
            if i % 3 == 0{
                x += 1
            }else if i % 3 == 1{
                y += 1
            }else{
                x -= 1
                y -= 1
            }
            pingping[x][y] = curnum
            curnum += 1
        }
    }
    
    //순서대로 0이 아니면 추가
    for i in 0..<n{
        for j in 0..<n{
            if pingping[i][j] != 0 { result.append(pingping[i][j]) }
        }
    }
    
    return result
}