[프로그래머스] 카펫 (Python)

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

코딩테스트 연습 - 카펫

 

1️⃣ yellow의 약수를 구해 divisor에 담아 주었다. (yellow의 너비와 높이)

2️⃣ 약수 리스트에서 하나씩 뽑아 너비*2+높이*2+4가 brown과 같으면(아래 이미지 참고) answer로 너비와 높이에 각각 2씩 더해 넣어주었다.

 

2a+2b+4 = brown

 

def solution(brown, yellow):
    answer = []
    divisor = []
    for num in range(1, yellow+1):
        if yellow%num == 0:
            if num >= yellow//num:
                divisor.append([num, yellow//num])
    for comb in divisor:
        if comb[0]*2+comb[1]*2+4 == brown:
            answer = [comb[0]+2, comb[1]+2]
    return answer