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

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

문제 링크

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

프로그래머스

문제 정답

brown	yellow	returns
10	2	[4, 3]
8	1	[3, 3]
24	24	[8, 6]

문제 이미지

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

---

문제 해설

일단 우리는 격자 갯수를 통해 너비를 알 수 있다.

그리고 다음 조건이 주어진 것을 알 수 있다.

• w길이는 h 길이와 같거나, 세로 길이보다 길다.

따라서 h를 1부터 키우면 h보다 작게 된다.

    for h in range(1,int(total**(1/2))+1):
        w = total // h

그리고 너비 조건을 확인한다. 

우리는 테두리 조건을 통해 갈색 갯수를 구할 수 있다.

• 2(w+h) - 4
  
  • 각 모서리가 중복으로 카운트 되기 때문
  • 이 값이 갈색 갯수랑 동일한지 확인한다/

테두리 조건이 확인되면 아래와 같이 [가로,세로]를 리턴한다

return [w,h]