프로그래머스 level2 [카펫] 파이썬

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

코딩테스트 연습 - 카펫

간단한 수학문제이다.

#프로그래머스 카펫
def solution(brown, yellow):
    for h in range(1,yellow+1):
        if yellow%h==0:
            w=yellow//h
            if w<h:
                break
            if (w+2)*2+2*h==brown:
                return [w+2,h+2]

 

더 좋은 풀이 (다른 분 풀이)

def solution(brown, red):
    for i in range(1, int(red**(1/2))+1):
        if red % i == 0:
            if 2*(i + red//i) == brown-4:
                return [red//i+2, i+2]