[프로그래머스][Python] 카펫

📃 문제 설명

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

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

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

제한사항

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

입출력 예

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

💻 문제 풀이

def solution(brown, yellow):                   
    for i in range(1, int(yellow**0.5)+1):      # 약수를 구하기 위해 sqrt(yellow) 값을 해준다.
        if yellow % i == 0:                     # 해당 값이 약수라면
            row, col = i, yellow//i             # row 에는 약수를 col 에는 곱했을 때 yellow값이 나오는 수를 넣어준다.
            if ((row+2)*2) + (col*2) == brown:  # 갈색은 노란색의 테두리이기 때문에 테두리 개수를 구하는 식이 brown 개수와 같다면
                return [col+2, row+2]           # row,col에 +2 해준 후 리턴 (col >= row)


print(solution(10,2))      # result : [4,3]
print(solution(8,1))       # result : [3,3]
print(solution(24,24))     # result : [8,6]

중요 포인트

• 약수를 구할 때 int(yellow**0.5))+1 까지 for문 돌리기.

비교적 쉬운 문제였다.

함수로 분리해서 짠 코드

처음에 함수로 분리해서 코드를 작성했지만, 굳이 함수로 분리해서 할 필요가 없을 것 같았다.

def get_divisor(num):                           # 약수 구하는 함수 list를 리턴
    divisor = []
    for i in range(1, int(num**0.5)+1):         
        if num % i == 0:
            divisor.append([i, num//i])
    return divisor

def solution(brown, yellow):
    yellow_list = get_divisor(yellow)           # yellow_list = 약수로 이루어진 2차배열

    for row, col in yellow_list:
        if ((row+2)*2) + (col*2) == brown:      # 갈색은 노란색의 테두리이기 때문에 테두리 개수를 구하는 식이 brown 개수와 같다면
            return [col+2, row+2]               # row,col에 +2 해준 후 리턴 (col >= row)