[프로그래머스 LV2] 카펫

---

개인적으로 수학적인 사고를 조금 더한다면 쉽게 풀 수 있는 문제라고 생각합니다.

문제

카펫 링크

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

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

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

제한사항

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

입출력 예

내 풀이

오랜만에 제 풀이를 보았는데, 만든 코드인데도 헷갈려서 시간이 걸렸네요…😅 yellow의 약수가 가로, 세로의 길이와 연관이 있다는 점을 이용하여 만들었습니다. 아쉬운 점이 있다면, 첫 번째 if 문을 만족하는 가로, 세로 들이 모두 brown 식을 만족하는지 확인했다는 점입니다. 또한, yellow가 상당히 큰 수라면, 두 번째 if 문뿐 아니라 처음 for 문을 돌리는 데도 시간이 오래 걸릴 것입니다. 차라리 brown 식을 만족했을 때, 바로 break 구문을 넣어서 시간을 단축시키는 것이 더 좋았을 것이라고 생각합니다.

def solution(brown, yellow):
    answer = []

    for i in range(1, (yellow + 1)) :
      if yellow % i == 0 :     # 가로, 세로는 정수이므로 i는 yellow의 약수여야 한다.
        x = max(((yellow // i) + 2), (i + 2))     # (yellow를 i로 나눈 몫 + 2) 혹은 (i + 2)가 가로와 세로이다.
        y = min(((yellow // i) + 2), (i + 2))
        if (x + y - 2) * 2 == brown :     # 단, 만족하는 가로, 세로는 다음 식을 만족해야 한다.
          answer.extend([x, y])
          return answer

다른 사람들의 풀이

아래는 간단한 이차방정식 문제로 회귀시켜 문제를 푼 코드입니다. 매번 느끼는 것이지만, 문제는 최대한 쉽게 풀어가는 것이 좋은 것 같습니다. 아래와 같은 생각이 바로 떠오르기까지 얼마나 걸릴지… 또 한번 생각하게 되는 코드였어요. 😂

import math
def solution(brown, yellow):
    ans=((brown-4)+math.sqrt((brown-4)**2-16*yellow))//4
    return [ans+2,yellow//ans+2]