프로그래머스 카펫

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

프로그래머스

 

처음에 해당 문제를 보고 입출력 예시를 보았습니다.

가장 먼저 떠오른 생각은 brown과 yellow의 합을 기준으로 그 약수들을 구하고,
그 중에서 (가로, 세로) 쌍의 곱이 brown + yellow가 되면서 조건을 만족하는 가장 작은 가로 값을 찾는 것이 답이 아닐까 하는 것이었습니다.

예를 들어, brown이 10이고 yellow가 2라면 합은 12이고,
12의 약수 쌍은 (1,12), (2,6), (3,4)와 같습니다.
이런 가정하에 문제를 접근했습니다.

def solution(brown, yellow):
    temp = brown + yellow
    ans = 1e9
    result = []
    for i in range(1,brown+1) :
        if temp % i == 0 :
            row = i 
            col = temp //i
        else:
            continue
        if ans > row + col:
            ans = row + col  
            result = [row,col]
    result.sort(reverse=True)
    return result

 

이러면 

 

반례를 찾았는데 

solution( 26, 10) # [12, 3]

 

기대값과 다르게 [6,6]이 나옵니다.

 

그래서 그림을 그렸는데 .. 

해당 수식으로 일반화 할 수 있었습니다 !

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