[Programmers] 카펫 - 42842

카펫.. 고민을 많이 한 문제.

 

Question

 

Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

 

 

Code

 

def solution(b, y):
    
    ys = [i for i in range(1,y+1) if y%i == 0]
    for i in ys[::-1]:
        if b == (i+2 + y//i)*2 : return [i+2, y//i+2]

 

now me

On my github