[Programmers] 카펫

 

 

문제 

 

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

 

Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.

Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

 

 

제한 사항

• 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
• 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
• 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

 

나의 코드 

def solution(brown, yellow):
    # a와 b의 조건 (a = 가로, b = 세로)
    # yellow = (a - 2) * (b - 2)
    # total = a * b
    total = brown + yellow
    array = [] # 가로 x 세로가 될 수 있는 경우의 수 모음
    for i in range(total):
        if i > 0:
            if total % i == 0:
                array.append(i)
                if i * i == total:
                    array.append(i)
    print(array)
    
    # 그중에서 두가지 조건을 충족하는 가로, 세로 길이 구하기
    for p in range(len(array)):
        for t in range(p+1,len(array)):
            if array[p] * array[t] == total:
                if (array[p] - 2) * (array[t] - 2) == yellow:
                    answer = [array[p],array[t]]
                    answer.sort(reverse=True)
                    return answer

분석 > 우선 전체 격자의 개수를 토대로 나올 수 있는 (가로 x 세로) 쌍을 구해 array에 append 하였다. 그 후 두 가지 조건을 충족하는 (가로 x 세로) 쌍을 구해 반환해주었다. 

 

효율적인 코드 

def solution(brown, yellow):
    total = brown + yellow
    for n in range(1, total+1):
        if total%n != 0:
            continue
        m = total//n
        if (n-2)*(m-2) == yellow:
            return sorted([n, m], reverse = True)

분석 > 전체적인 흐름 자체는 내가 짠 코드와 크게 다르지 않았다. 다만 나는 나올 수 있는 모든 (가로 x 세로) 쌍을 모두 구해주었는데, 이 코드는 continue를 사용해 가장 마지막에 입력된 n값을 기반으로 m을 구해주었고 이 두 값이 조건을 충족하면 값을 반환해주는 방식으로 코드를 전개하였다. 이렇게 굳이 필요하지 않는 부분은 간결하게 처리하고. 최대한 효율적으로 코드를 짜는 법을 학습할 필요성을 느꼈다.