프로그래머스 연습문제 - 카펫 (level2)

j_wisdom_h·2022년 11월 21일

level2 스터디 카펫 코딩테스트 코테 프로그래머스

CodingTest

목록 보기

14/58

프로그래머스 연습문제 - 카펫 (level2)

문제설명

제한사항 & 입출력 예

My Solution

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

#위 코드를 2줄로 쓴다면 ( list comprehension )
#하지만 효율성은 위 코드가 더 좋다.
def solution(brown, yellow):
    total = brown + yellow 
    return  [[w,total//w] for w in range(1,total+1) if total % w ==0 and w >= total/w and (w-2)*(total//w - 2) == yellow][0]

와 오늘은 문제해석을 잘못해서 엄청 해맸다.

처음에는 아무생각없이 테스트케이스만 보고 둘을 더한값을 곱으로 표현했을때 가장 적은 격차가 있는 [w,h]을 반환했다.
예를 들어,
brown : 10, yellow: 2 라면 이 둘의 합은 12이다.
width와 height를 곱해서 12가 되는 수 중 서로 차이가 가장 작은 것은 [4,3] (w>=h 이므로) 이라고 구했다.
지금 생각하면 어이가 없다 하하하.

def solution(brown, yellow):
    answer = []
    total = brown + yellow 
    for i in range(1,total+1):
        if total % i ==0 and i >= total/i :
            answer= [i,int(total/i)]  
            return answer