[알고리즘] 프로그래머스(삼각 달팽이 - Swift)

1. 삼각 달팽이

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

2. 제한조건

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

3. 입출력 예

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]

4. 정답코드

import Foundation

func solution(_ n:Int) -> [Int] {
    var result = [[Int]](repeating: [Int](repeating: 0, count: n), count: n)
    var x = -1
    var y = 0
    var num = 1

    for i in 0..<n {
        for _ in 0..<(n - i) {
            if i % 3 == 0 { // Down
                x += 1
            } else if i % 3 == 1 {  // Right
                y += 1
            } else {    // Up
                x -= 1
                y -= 1
            }
            
            result[x][y] = num
            num += 1
        }
    }

    return result.flatMap { $0 }.filter { $0 != 0 }
}