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

프로그래머스

---

성공 풀이

func solution(_ n:Int) -> [Int] {
  var (y,x) = (n-1,0) // 좌표 설정
  var count = n // 대입할 숫자
  let sum = (1...n).reduce(0,+) // 1~n까지 각각 의 합 (break할 때 필요)
  var arr = (0..<n).map { // 배열의 윤곽 초기화
    [$0+1] + Array(repeating: -1, count: n-$0-1) + Array(repeating: 0, count: $0)
  }
  
while true {
  
  while x < n-1 && arr[y][x+1] == 0 { // 오른쪽으로
    count += 1
    arr[y][x+1] = count
    x += 1
  }
  
  while y > 0 && arr[y-1][x] == 0 { // 위쪽으로
    count += 1
    arr[y-1][x] = count
    y -= 1
  }
  
  while y < n-1 && arr[y+1][x-1] == 0 { // 대각선 아래로
    count += 1
    arr[y+1][x-1] = count
    y += 1; x -= 1
  }
  
  if count == sum { break } else { continue } // count가 sum과 같다면 break 아니면 continue
}
  arr.forEach { print($0) }
  return arr.flatMap { $0 }.filter { $0 > 0 }
}

풀이 키워드

구현

---