[Python] 카펫

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

프로그래머스

# 카펫

def solution(brown, yellow):
  
    answer = []
    total = brown + yellow   # a * b = total
    
    for b in range(1,total+1):
        
        if total / b == int(total/b):            # total / b = a, 정수여야 한다.
            a = total / b
            if a >= b:                      # a >= b
                if 2*a + 2*b == brown + 4:  # 2*a + 2*b = brown + 4 
                    return [a,b]
            
    return answer