방향 전환은 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
}