[프로그래머스-Lv2] 카펫 / Python

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

코딩테스트 연습 - 카펫

import sys

input = sys.stdin.readline

brown, yellow = map(int, input().split())


def solution(brown, yellow):
    for i in range(1, yellow + 1):
        if yellow % i == 0:
            temp = yellow // i
            if i * 2 + temp * 2 + 4 == brown:
                return [temp + 2, i + 2]


print(solution(brown, yellow))

 

해설

일단, 노란 블럭의 가로와 세로 길이를 구해줘야 한다. 위 코드에서 i는 가로, temp는 세로이다.

갈색 블럭의 개수는 노란 블럭의 가로길이*2 + 세로길이*2 + 4(모서리)이다.

노란 블럭의 가로와 세로 길이를 구하고 위 식이 갈색 블럭의 개수와 맞다면 [세로길이+2, 가로길이+2]를 리턴한다.