[프로그래머스] 완전탐색 - 카펫 (level.2)

 

문제링크

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

코딩테스트 연습 - 카펫

 

나의 풀이

일단 brown과yellow의 합의 약수만 한변의 길이가 될 수 있다는 것을 캐치해야했다.
그리고, 규칙을 살펴보면, 문제의 그림과 같이 yellow가 brown에 모두 둘러쌓여있으려면 각 변의 -2한 것들의 곱이 yellow의 갯수와 같아야함을 알 수 있다. 그대로 생각힌대로 코드를 구현했다.

def solution(brown, yellow):
    answer = [] 
    div = [] # 약수를 넣을거임
    total = brown + yellow
    for i in range(3, total+1):
        if total % i == 0:
            div.append(i)
    for d in div:
        n = total //d
        if (d-2)*(n-2) == yellow :
            answer.append(n)
            answer.append(d)
            answer.sort(reverse = True)
            break
        else:
            pass
    return answer

 

다른사람풀이

https://programmers.co.kr/learn/courses/30/lessons/42842/solution_groups?language=python3 

프로그래머스

def solution(brown, red):
    for i in range(1, int(red**(1/2))+1):
        if red % i == 0:
            if 2*(i + red//i) == brown-4:
                return [red//i+2, i+2]

코드가 상당히 간결하다. 하지만 마지막에 결국 +2씩 해놓은 것을 보면, 나와 같은 규칙을 찾은것을 확인할 수있다. 코드를 참고해서 조금 더 간결한 코드를 짤 수 있도록 노력해보자!