링크: https://school.programmers.co.kr/learn/courses/30/lessons/86491
문제 설명
명함 지갑을 만드는 회사에서 지갑의 크기를 정하려고 합니다. 다양한 모양과 크기의 명함들을 모두 수납할 수 있으면서, 작아서 들고 다니기 편한 지갑을 만들어야 합니다. 이러한 요건을 만족하는 지갑을 만들기 위해 디자인팀은 모든 명함의 가로 길이와 세로 길이를 조사했습니다.
아래 표는 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
문제 예시와 같습니다.
입출력 예 #2
명함들을 적절히 회전시켜 겹쳤을 때, 3번째 명함(가로: 8, 세로: 15)이 다른 모든 명함보다 크기가 큽니다. 따라서 지갑의 크기는 3번째 명함의 크기와 같으며, 120(=8 x 15)을 return 합니다.
입출력 예 #3
명함들을 적절히 회전시켜 겹쳤을 때, 모든 명함을 포함하는 가장 작은 지갑의 크기는 133(=19 x 7)입니다.
문제풀이
문제는 모든 명함을 담을 수 있는 명함 지갑의 크기를 구하는 것
- 기본 → 가로와 세로의 max 를 구하는 것.
- 하지만? → 가로로 눕히거나 세로로 돌려 수납할 수 있다.
- 가로와 세로로 돌릴 때의 기준을 어떻게 세우지?
로직:
- 가로와 세로 각각 max 를 비교한다.
- 비교해서 더 큰쪽은 그대로, 작은 쪽은 가로, 세로 순서 뒤바꾸기
- 그리고 다시 가로, 세로 max 를 구해서 그 값들을 곱해 답 구하기
def solution(sizes):
width = []
height = []
for w, h in sizes:
width.append(w)
height.append(h)
max_width = max(width)
max_height = max(height)
if max_width > max_height:
width[height.index(max_height)], height[height.index(max_height)] = height[height.index(max_height)], width[height.index(max_height)]
max_height = max(height)
elif max_height > max_width:
height[width.index(max_width)], width[width.index(max_width)] = width[width.index(max_width)], height[width.index(max_width)]
max_width = max(width)
return max_width*max_height이렇게 했을 때 첫 번째 입출력 예시는 통과했지만 나머지 두 개는 오답이 나왔다.
- 다시 로직을 생각해봤다.
- 나는 계속 각 요소의 가로와 세로 위치를 지키면서 width, height 리스트에 추가하려고 했다.
- 하지만 그럴 필요 없이 sizes 의 요소를 바로 비교해서 큰 값은 큰 값 끼리, 작은 값은 작은 값 끼리 리스트에 넣고 거기서 max 끼리 곱하면 되는 쉬운 문제 아닌가??
def solution(sizes):
large_arr = []
small_arr = []
for i in sizes:
large_arr.append(max(i))
small_arr.append(min(i))
return max(large_arr) * max(small_arr)성공!
다른사람 풀이
def solution(sizes):
return max(max(x) for x in sizes) * max(min(x) for x in sizes)출처: https://school.programmers.co.kr/learn/courses/30/lessons/86491/solution_groups?language=python3
내가 작성한 6줄의 코드를 이렇게 간단히 한 줄로 축약 가능하다니.. 역시 아직 갈 길이 먼 거 같다.
