[프로그래머스][Lv.2][Python] 카펫

1. 문제

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

프로그래머스

 

2. 접근방식

• 주어진 brown과 yellow의 개수는 해당 색 부분의 넓이와 같다.
• 가로를 x, 세로를 y라고 할 때, 주어진 brown과 yellow의 조합으로 xy와 x+y를 알 수 있다.
• 간단한 이차방정식을 만들고 이후는 완전탐색으로 값을 찾음

 

3. 코드

def solution(brown, yellow):
    answer = []
    
    x_y = int(brown / 2) + 2
    xy = brown + yellow
    
    for x in range(1, x_y):
        for y in range(1, x_y):
            if (x + y == x_y) and (x*y == xy):
                if x >= y:
                    return [x, y]
        
    return answer

 

4. 결과