[Python] 카펫 - level 2

https://programmers.co.kr/learn/courses/30/lessons/42842?language=python3 

코딩테스트 연습 - 카펫

 

 

 

 

넓이는 가로x세로 라서

yellow에 제곱근을 해주어 for문을 실행시켰다.

 

넓이=면적이기때문에 아래와 같이 가능하다.

 

또는 근의 공식을 써서 풀 수 있다는데 대단한 사람들이 많다.

난 이렇게 해야지...

 

 

 

def solution(brown, yellow):    # 가로x세로 = 넓이(=면적)(=brown+yellow)
    answer = []

    # 가로 세로는
    for i in range(int(yellow**(1/2)), 0, -1):
        width = yellow//i + 2   # yellow의 가로 길이
        height = i + 2          # yellow의 세로 길이
        if width >= height and brown == height*2 + (width-2) * 2:
            answer = [width, height]
            break

    return answer