[프로그래머스] [level2] 카펫 - python

문제 링크

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

코딩테스트 연습 - 카펫

 

문제 풀이 

def solution(brown, yellow):
  # i*2 , j*2 의 합은 brown - 4 
  # i * j 곱은 yellow    
  # 결과값은 i+2, j+ 2 
  for i in range(1, brown):
    for j in range(1, brown):
      if i*2 + j*2 == brown-4 and i*j == yellow:
        # 가로의 길이가 더 길기 때문에 
        if i > j:
          return [i+2,j+2]
        else:
          return [j+2,i+2]