[파이썬/알고리즘] 프로그래머스 - 최소직사각형

[문제]

명함 지갑을 만드는 회사에서 지갑의 크기를 정하려고 합니다. 다양한 모양과 크기의 명함들을 모두 수납할 수 있으면서, 작아서 들고 다니기 편한 지갑을 만들어야 합니다. 이러한 요건을 만족하는 지갑을 만들기 위해 디자인팀은 모든 명함의 가로 길이와 세로 길이를 조사했습니다.

아래 표는 4가지 명함의 가로 길이와 세로 길이를 나타냅니다.

명함 번호 가로 길이 세로 길이
1 60 50
2 30 70
3 60 30
4 80 40
가장 긴 가로 길이와 세로 길이가 각각 80, 70이기 때문에 80(가로) x 70(세로) 크기의 지갑을 만들면 모든 명함들을 수납할 수 있습니다. 하지만 2번 명함을 가로로 눕혀 수납한다면 80(가로) x 50(세로) 크기의 지갑으로 모든 명함들을 수납할 수 있습니다. 이때의 지갑 크기는 4000(=80 x 50)입니다.

모든 명함의 가로 길이와 세로 길이를 나타내는 2차원 배열 sizes가 매개변수로 주어집니다. 모든 명함을 수납할 수 있는 가장 작은 지갑을 만들 때, 지갑의 크기를 return 하도록 solution 함수를 완성해주세요.

 

[풀이]

def solution(sizes):
    max_width = 0
    max_height = 0
    
    for size in sizes:
        w, h = size
        if w < h:
            w, h = h, w
        max_width = max(max_width, w)
        max_height = max(max_height, h)
        
    return max_width * max_height

가로, 세로의 크기를 비교하여 큰 값을 가로에 몰아넣는다.

그러고나서 가로 중에서 가장 큰 값과 세로 중에서 가장 큰 값을 곱해주면 지갑의 크기를 구할 수 있다.

 

이렇게 되는 이유는, 회전이 가능하기 때문에 가로 70 세로 30은 가로 30 세로 70이랑 똑같다.

그래서 가로 세로 중 큰 값을 한 쪽(가로)에 몰아넣고 여기서 가장 큰 값과 나머지 한 쪽(세로)에서 가장 큰 값을 곱해주면 된다.

 

max_width = 0

max_height = 0

가로, 세로 중 가장 큰 값을 찾기 위해서 변수를 선언해준다.

 

for size in sizes:
        w, h = size

함수에서 sizes로 받은 크기를 w, h 변수로 받는다.

 

 if w < h:
            w, h = h, w

만약 세로가 가로보다 크면 순서를 바꿔준다.

 

max_width = max(max_width, w)
max_height = max(max_height, h)

반복문을 돌고 비교를 하면서 가장 큰 가로를 max_width 변수에, 가장 큰 세로를 max_height 변수로 받는다.
       

return max_width * max_height

최종적으로 둘을 곱해서 지갑의 최소 크기를 반환한다