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

비얌·2022년 4월 19일

알고리즘 프로그래머스

알고리즘

목록 보기

9/17

프로그래머스 1단계 문제인 '최소 직사각형' 문제를 풀어보았다.

문제

https://programmers.co.kr/learn/courses/30/lessons/86491

문제 설명

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

가장 긴 가로 길이와 세로 길이가 각각 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 이하인 자연수입니다.

입출력 예

내 풀이(틀림)

너비와 높이를 all이라는 리스트에 한번에 모아서 내림차순으로 정렬한 후 중간에 있는 값과 가장 큰 값을 곱해서 명함의 최소 크기를 구하려고 했다.

(세로 중 가장 큰 수와 가로 중 가장 큰 수를 곱하려고 했음)

의도는 정답과 비슷했으나, 세로는 세로끼리, 가로는 가로끼리 계산해야 했다. 그래서 all 한군데로 모은 것처럼 하면 안되고 가로, 세로 각각 넣어서 계산해줘야 했다.

예를 들어 (10, 20), (20, 30), (30, 90)이 있다고 치자. 그러면 명함의 최소 크기는 30 * 90이 될 것이다. 하지만 가로세로에 개의치않고 정렬하여 반으로 가른다면 10, 20, 20, 30, 30, 90이 되어 20 * 90이 답이 될 것이다.

def solution(sizes):
    all = []
    for i in range(len(sizes)):
        all.append(sizes[i][0])
        all.append(sizes[i][1])
    all.sort()
    return (all[-1] * all[len(sizes)-1])

다른 풀이

def solution(sizes):
    w = []
    h = []
    for i in range(len(sizes)):
        if sizes[i][0] >= sizes[i][1]:
            w.append(sizes[i][0])
            h.append(sizes[i][1])
        else:
            h.append(sizes[i][0])
            w.append(sizes[i][1])

    return max(w) * max(h)