[Python/Programmers] 카펫

문제

1. 난이도: 프로그래머스 레벨 2

2. 문제 요약

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

Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.

Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항

• 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
• 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
• 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

3. 문제 핵심

• 모든 경우의 수를 다 확인해봐야 -> 완전 탐색
  ex) 4자리의 암호로 구성된 자물쇠를 풀려고 시도할 때,0000 ~ 9999까지 모두 시도(무조건 답은 나옴)

내 코드

import math
def solution(brown, yellow):
    for yHeight in range(1, yellow):
        yWidth = yellow / yHeight
        
        if yWidth != int(yWidth):
            continue
            
        elif (yWidth * yHeight!= yellow) or (yWidth < yHeight):
            continue
            
        else:
            bWidth = yWidth + 2
            bHeight = yHeight + 2
            if bWidth * bHeight - yellow == brown:
                return [bWidth, bHeight]
    if yellow ==1:
        return [3,3]

이상적인 코드

def solution(brown, yellow):
    total = brown + yellow
    for i in range(total, 2, -1): ## i는 가로
    	if total % i == 0:
        	j = total // i ## j는 세로
            
            if yellow == (i-2)*(j-2):
            	return [i, j]