[Algorithm] 카펫

[hwangJi-dev]

💬 문제

https://school.programmers.co.kr/learn/courses/30/lessons/42842

💬 Idea

• 노란색을 직사각형으로 채울 수 있는 가로, 세로를 찾기 위해 반복문을 돌면서 나머지가 0인 경우를 찾는다.
• 만약 (yx * 2) + (i * 2) + 4 가 브라운의 값과 같다면 target이므로 break한다.

💬 풀이

import Foundation

func solution(_ brown:Int, _ yellow:Int) -> [Int] {
    var res: [Int] = []
    
    if yellow == 1 {
        return [3, 3]
    }
    
    for i in 1...yellow / 2 + 1 {
        if yellow % i == 0 {
            let yx = yellow / i
            let ans = (yx * 2) + (i * 2) + 4
            
            if ans == brown {
                res = [yx + 2, i + 2]
                break
            }
        }
    }
    
    return res
}