프로그래머스 - 카펫(Swift)

프로그래머스 - 카펫(Swift)

 

 

 문제 설명

 

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

프로그래머스

 

 나의 풀이

 

처음 문제를 보고 읽으면서 문제가 담고있는 의미를 생각하지 않았기 때문에 오래 걸렸다.

핵심은 두 가지였다.

 

1. 테두리 1줄은 갈색으로 이루어져 있다.

2. 가로는 세로보다 같거나 크다.

 

테두리가 겉으로 1줄만 있기 때문에 가로-2, 세로-2는 항상 노란색이라는 것을 알 수 있다.

그리고 가로가 세로보다 크기 때문에 순차적으로 탐색할 때 몫을 w로 주고 탐색되는 값을 h로 줌으로써 넓이에 대해 약수를 구할 수 있다.

 

import Foundation

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