완전탐색Lv1최소직사각형

문제

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

아래 표는 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 이하인 자연수입니다.

풀이

• 배열을 큰 값이 0번 인덱스에 오도록 정렬
• 후에 0번 인덱스중 max 값과 1번 인덱스의 max값 도출
• 두 max값을 곱한다

코드

def solution(sizes):
    # 가로 세로는 변경할 수 있다.
    # 먼저 가장 긴 길이를 찾아라. 아마 둘중에 큰 값을 앞에다 놓고 max 들을 구하면 되지 않을까?
    answer = 0
    newArr = [0]*len(sizes)
    for i in range(len(sizes)):
        if sizes[i][0] >= sizes[i][1]:
            newArr[i] = (sizes[i][0],sizes[i][1])
        else: 
            newArr[i] = (sizes[i][1],sizes[i][0])
    
    maxCol = 0
    maxRow = 0

    for col, row in newArr:
        if col > maxCol:
            maxCol = col
        if row > maxRow:
            maxRow = row
    answer = maxCol * maxRow
    return answer

if __name__ == "__main__":
    sizes = [[10, 7], [12, 3], [8, 15], [14, 7], [5, 15]]
    print(solution(sizes))

배운것

• 멋진 코드. 1줄로 가능하다

def solution(sizes):
    return max(max(x) for x in sizes) * max(min(x) for x in sizes)

큰것들 중에 큰것과 작은것들 중에 큰것을 곱한 것의 max값을 찾는다.

• SUM 함수

solution = lambda sizes: max(sum(sizes, [])) * max(min(size) for size in sizes)

empty list를 전달해서 sizes의 총합이 아니라 sizes의 모든 요소가 담긴 1차원 list로 변환