[python] 프로그래머스_카펫 (완전 탐색)

문제

풀이

카펫 그림을 보고 가로길이 = a, 세로길이 = b 로 두고 조건에 부합하는 식을 만들고 for 문 돌면서 if 문으로 조건에 맞는 경우를 찾으면 된다.

 

def solution(brown, yellow):
  total  = brown + yellow
  for b in range(1, total+1):
    if total/b % 1 == 0:
      a = total/b
      if a>=b:
        if ((a-2)*(b-2) == yellow) and (2*a + 2*b -4 == brown):
          return [int(a),int(b)]