[프로그래머스] 카펫

관련 문제

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

프로그래머스

 

코드

def solution(brown, yellow):
    carpet_total = brown + yellow
    answer = []
    for y in range(3, carpet_total):
        x = carpet_total / y
        if x < y:
            break
        else:
            if (x-2) * (y-2) == yellow :
                answer = [x, y]
                break
    return answer

 

해결 포인트

x, y의 비율을 구한 뒤 내부 영역인 (x-2) * (y-2) 값이 yellow랑 일치하는지 확인하면 된다.