프로그래머스 / 카펫 / Python

S.T.Lee 2021. 12. 22. 21:08

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

코딩테스트 연습 - 카펫

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다. Leo는 집으로 돌아와서 아까 본 카펫의 노란색과

programmers.co.kr

def solution(brown, yellow):
    div_list = []
    color_sum = brown + yellow
    for number in range(1,color_sum): #가로, 세로 조합 찾기
        if int(color_sum/number) <= number: # 조건 : 가로는 세로 길이와 같거나 길다
            if color_sum % number == 0:
                div_list.append([number, int(color_sum/number)])
    for div in div_list:
        if (div[0]-2)*(div[1]-2) == yellow: #노랑색 또한 사각형 형태를 띈다
            return div