1️⃣ 문제
문제 설명
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.
Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.
Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.
제한사항
- 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
- 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
- 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.
입출력 예
brown | yellow | return |
---|---|---|
10 | 2 | [4, 3] |
8 | 1 | [3, 3] |
24 | 24 | [8, 6] |
2️⃣ 풀이 방법
갈색 줄은 무조건 1px이니까
카펫 w = 노란색 w + 2
카펫 y = 노란색 y + 2
격자의 수는 w * h 이므로, 노란색 w,h를 만들 수 있는 모든 경우의 수에 대해, 대응하는 카펫 w,h를 설정하고 맞다면 리턴
- 판별하는 방법은 갈색 격자 수 == 카펫의 둘레(2*(w+h)) - 4
3️⃣ 코드
import Foundation
func isRigthSize(_ size: [Int], _ brown: Int) -> Bool {
let sumWH = size.reduce(0, +)
return brown == 2 * sumWH - 4
}
func solution(_ brown:Int, _ yellow:Int) -> [Int] {
var carpet = [Int](repeating: 0, count: 2) // 카펫 w,h
for i in (1...yellow).reversed() {
if yellow % i == 0 {
carpet[0] = i + 2
carpet[1] = yellow/i + 2
if isRigthSize(carpet, brown) { return carpet }
}
}
return []
}