[프로그래머스] 카펫 (코딩테스트 연습 | 완전탐색)

문제

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

프로그래머스

---

아이디어

✅ '테두리 (Brown)이 1줄이다'

• 즉, yellow의 공약수 두 수 a, b 를 구하고 (a * b = yellow)
• (a + 2) * (b + 2) = (총 직사각형 넓이) 인 경우를 구해 b+2, a+2를 반환한다

 

✅ 공약수는 어떻게 구하는가?

• for문을 이용해 1부터 yellow로 탐색

---

코드

import math

def solution(brown, yellow):
    answer = []
    
    # 전체 넓이
    area = brown + yellow
    
    # 공약수 구하기
    for i in range(1, yellow+1):
        if yellow % i == 0:
            if (i + 2) * (yellow/i + 2) == area:
                return [yellow/i + 2, i+2]