가능한 행, 열을 통해 전체 노드의 개수를 카운트한다. 노드의 개수에서 바깥 테두리를 뺀 안 쪽 노란색 노드의 개수가 주어진 노란색 개수와 맞다면 성공.
import Foundation
func solution(_ brown:Int, _ yellow:Int) -> [Int] {
var brown = (brown + 4) / 2
var answer = [Int]()
for row in 1...(brown / 2) {
let col = brown - row
let boxCount = row * col
if boxCount - (row * 2) - (col * 2) + 4 == yellow {
answer = [col, row]
break
}
}
return answer
}