[프로그래머스] 완전탐색 - 카펫 (Level 2)

카펫

Python Code

def divisors(a):    #3보다 크고 가로가 세로보다 길거나 같은 약수 쌍 구하기
    result = []
    for i in range(3, int(a/2)):
        if a%i == 0 and i>=int(a/i):
            result.append([i, int(a/i)])
    return result

def solution(brown, yellow):
    answer = []
    lst = divisors(brown+yellow)
    for i in lst:
        if brown == 2*(i[0]+i[1])-4:
            answer = i
    return answer