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

카펫

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

프로그래머스

문제

코드

• 가로: a, 세로 b라고 하면 2a - 2b -4 = 갈색 카펫의 개수
• 방정식에 따라서 구현

def solution(brown, yellow):
    answer = []
    total = brown + yellow

    for b in range(1, total + 1):
        if (total / b) % 1 == 0:
            a = total / b
            if a >= b:
                if a*2 + b*2 == brown + 4:
                    return [a, b]