[프로그래머스] 카펫

문제

프로그래머스 카펫

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

프로그래머스

 

코드

def solution(brown, yellow):
    for i in range(5000 + 1):   # 세로길이
        for j in range(5000 + 1):   # 가로길이
            if i * j == brown + yellow and i + j - 1 == brown//2 + 1:
                return [j, i]

 

풀이

• 완전탐색이라는 키워드에 맞게 풀려고 최대한 노력했다. 
• 모든 경우의 수를 탐색하며 조건에 맞는 값을 반환하도록 했다.

for i in range(5000 + 1):   # 세로길이
        for j in range(5000 + 1):   # 가로길이

갈색이 최대로 5000 까지 들어갈 수 있으니 반복문을 5000 까지만 돌게 한다.

 

if i*j == brown + yellow and i+j-1 == brown//2 + 1:

가로 * 세로 (= 넓이) 가 갈색, 노랑 타일들의 개수합과 같고

둘레길이의 반이 갈색타일//2 + 1 과 같다면

 

return [j,i]

[가로길이, 세로길이]를 반환한다.

내가 쓴 코드의 이중 포문이 돌아가는 방식은 

i = 0 일때 j는 0~ 5000 까지돈다.

i = 1 일때 j는 0~ 5000 까지돈다.

i = 2 일때 j는 0~ 5000 까지돈다.

...

i = 5000 일때 j는 0~ 5000 까지돈다.

이런 방식이기 때문에 문제 조건 중 "카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다" 라는 조건을 충족시키기 위해서 특별히  반환 순서를 j, i 순서로 반환한다.

 

새롭게 배운점 / 느낀점 

• 개인적으로 제일 좋아하는 알고리즘이어서 즐겁게 풀었다. 
• 완전탐색 과정에서 조건문을 어떻게 걸어야지 효율적인 코드가 나올지 많이 고민하였다.
• 완전탐색 자체가 단순해서 그런지 효율성을 극대화한 조건문이 떠오르지 않았다.

혹시 좋은 조건문 아이디어가 있으신분은 댓글로 공유 바랍니다! :)