[알고리즘 풀이] [python] 프로그래머스, '카펫'

<문제 설명>

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 빨간색으로 칠해져 있고 모서리는 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

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

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

제한사항

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

---

<나의 풀이> --- 3개의 테스트 케이스가 Fail...

접근을 갈색 격자기준으로 경우의 수를 분류해 사이즈 크기가 빨간색 격자크기와 동일한 경우의 최댓값을 찾도록 했다.

(갈색->빨간색)

def solution(brown, red):
    sum = brown + red
    temp=[]
    for i in range(3,sum):
        if sum % i == 0 and i >= sum//i and sum//i > 2:
            temp.append((i, sum//i))
    
    max_judge=[]
    print(temp)
    for x, y in temp:
        value=(x - 1)*(y - 1)
        max_judge.append(value)
    idx = max_judge.index(max(max_judge))
    return list(temp[idx])

<다른 사람 참조한 풀이>

다른 사람 코드를 보았을 때, 접근은 빨간색 격자 기준으로 프로그램을 짰더라...

그리고 갈색 격자 갯수와 동일할 때 값을 반환한다.

def solution(brown, red):
    for i in range(1, int(red ** 0.5)+1 ):
        if red % i == 0:
            if 2*(i+1) + 2 * ( (red//i)+1 ) == brown:
                return [red//i + 2,i+2]