[프로그래머스] [Level 2] [완전 탐색] 카펫

카펫

코딩테스트 연습 - 카펫

 

• 내 풀이
  • r과 c를 곱하여 brown+yellow가 되는 값 중 r,c가 정수인 값 구함

def solution(brown, yellow):
    total=brown+yellow
    answer = []
    for y_r in range(yellow,0,-1):
        r=y_r+2
        if total%r!=0: 
            continue
        c=total//r
        y_c=c-2
        if (y_r+y_c)*2+4==brown:
            answer.append(r)
            answer.append(c)
            break
    return answer

 

→ 리스트 따로 넣을 필요 없이 바로 return [r,c] 해도 됨