Cohe

프로그래머스 최소직사각형 본문

개발 이모저모/코딩 테스트 공부

프로그래머스 최소직사각형

코헤0121 2024. 9. 28. 12:54
728x90

최소직사각형 문제

문제

문제 설명

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

아래 표는 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 함수를 완성해주세요.


제한사항

  • sizes의 길이는 1 이상 10,000 이하입니다.
    • sizes의 원소는 [w, h] 형식입니다.
    • w는 명함의 가로 길이를 나타냅니다.
    • h는 명함의 세로 길이를 나타냅니다.
    • w와 h는 1 이상 1,000 이하인 자연수입니다.

입출력 예

sizes result
[[60, 50], [30, 70], [60, 30], [80, 40]] 4000
[[10, 7], [12, 3], [8, 15], [14, 7], [5, 15]] 120
[[14, 4], [19, 6], [6, 16], [18, 7], [7, 11]] 133

입출력 예 설명

입출력 예 #1
문제 예시와 같습니다.

입출력 예 #3
명함들을 적절히 회전시켜 겹쳤을 때, 모든 명함을 포함하는 가장 작은 지갑의 크기는 133(=19 x 7)입니다.

입출력 예 #2
명함들을 적절히 회전시켜 겹쳤을 때, 3번째 명함(가로: 8, 세로: 15)이 다른 모든 명함보다 크기가 큽니다. 따라서 지갑의 크기는 3번째 명함의 크기와 같으며, 120(=8 x 15)을 return 합니다.


풀이

일단 전혀 모르겠어서!! (사실 전에도 풀려고 시도했지만 포기한 문제임) 그냥 풀이부터 보려고 했지만 한 문장씩 차례로 시도해봤다.

가장 긴 가로 길이와 세로 길이가 각각 80, 70이기 때문에 80(가로) x 70(세로) 크기의 지갑을 만들면 모든 명함들을 수납할 수 있습니다. 이 문장을 따라 가로 최대길이와 세로 최대길이를 곱하면 되는 것으로 생각해 봤었다.

def solution(sizes):
    answer = 0
    sizes[1].sort()
    max_second = max(size[1] for size in sizes) # 인공지능 코드!!
    return sizes[-1][0] * max_second

역시나 예상했던 대로 아니었다.

하지만 2번 명함을 가로로 눕혀 수납한다면 80(가로) x 50(세로) 크기의 지갑으로 모든 명함들을 수납할 수 있습니다. 이때의 지갑 크기는 4000(=80 x 50)입니다.
이제는 위 문장을 생각해야 할 차례였다

음... 사실 이미 몇 번 검색해봤어서... sizes 라는 배열을 1차적으로 정렬하고, 그 수 중 가장 큰 것들끼리 곱해 answer를 만들어보자고 생각했다


def solution(sizes):
    sizes = [sorted(size, reverse=True) for size in sizes]

    max_first = max(size[0] for size in sizes)
    max_second = max(size[1] for size in sizes)
    return max_first * max_second

합격!!

배운점

  1. 리스트 컴프리헨션:

    • [sorted(size, reverse=True) for size in sizes]와 같은 리스트 컴프리헨션을 사용
    • 이는 각 하위 리스트를 순회하면서 정렬함
  2. max() 함수와 제너레이터 표현식:

    • max(size[0] for size in sizes)와 같은 표현을 사용