[프로그래머스 Programmers][Python] 카펫

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

코딩테스트 연습 - 카펫 | 프로그래머스

문제를 잘 보면, red가 W*H라고 할때, 전체 박스 크기는 (W+2)*(H+2)와 같고, 이는 brown+red의 값과 같다.

 

그때, red의 가로-세로가 될 수 있는 약수 모음을 만들고(divisor_list) 그 중에서  (W+2)*(H+2) == brown+red 를 만족하는 값을 찾는다.

 

 

 

def divisor_list(num):
    ret = []
    for i in range(1, num+1):
        if num%i==0:
            ret.append(i)
    
    return ret

def solution(brown, red):
    answer = []
    d = divisor_list(red)
    for i in range(len(d)//2+1):
        if 2*d[i] + 2*(d[-i-1]+2) == brown:
            answer.append(d[-i-1]+2 )
            answer.append(d[i]+2 )
            
            break


    return answer