[Level 2] 카펫 (Python)

코딩테스트 연습 - 카펫

 

import math

def solution(brown, yellow):
    total = brown + yellow
    
    m, n = [], []
    
    for i in range(1, math.floor(math.sqrt(total)) + 1):
        if divmod(total, i)[1] == 0:
            n.append(i)
            m.append(int(total/i))
    
    for i in range(len(n)):
        if 2*(m[i] + n[i]) - 4 == brown:
            return(m[i], n[i])

 

▷ 갈색 격자의 수(Brown), 노란색 격자의 수(Yellow), 가로의 길이(m), 세로의 길이(n)에 대한 관계식은 다음과 같다.

 

(1) Brown + Yellow = m × n

(2) Brown = 2 × (m + n) - 4

 

▷ (1)을 만족하는 주어진 갈색 격자의 수와 노란색 격자의 수를 만족하는 가로와 세로의 길이를 찾는다.

 

▷ (1)을 만족하는 경우 중 (2)를 만족하는 경우가 있으면 이를 결과로 출력한다.

 

▶ []는 list를 할당한다.

 

▶ divmod 함수의 첫 번째 인자는 나누는 값, 두 번째 인자는 나눌 값을 의미한다. 결과는 몫과 나머지가 포함된 튜플로 출력된다.

 

▶math 라이브러리는 올림이 가능한 ceil 함수와 내림이 가능한 floor 함수를 포함하고 있다.