문제 설명
S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.
물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.
부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.
제한 사항
-
d는 부서별로 신청한 금액이 들어있는 배열이며, 길이(전체 부서의 개수)는 1 이상 100 이하입니다.
-
d의 각 원소는 부서별로 신청한 금액을 나타내며, 부서별 신청 금액은 1 이상 100,000 이하의 자연수입니다.
-
budget은 예산을 나타내며, 1 이상 10,000,000 이하의 자연수입니다.
입출력 예
| d | budget | result |
|---|---|---|
| [1,3,2,5,4] | 9 | 3 |
| [2,2,3,3] | 10 | 4 |
풀이
def solution(d, budget):
d.sort()
budget_sum = 0
for i in range(len(d)):
budget_sum += d[i]
if budget_sum > budget:
return i
return len(d)해결 과정
-
가장 적은 예산을 필요로 하는 것부터 사주기
이번 문제는 그리디(탐욕법)을 적용시킬 필요가 있어 보였다. 예산은 한정되어 있고, 가장 효율적으로 사용하는 것이 아니라 가장 많은 부서의 요청을 들어주는 것이 문제에서 요구하는 것이기 때문이다.가장 많은 부서의 요청을 들어주기 위해서는 가장 적은 예산을 필요로 하는 곳부터 요청을 들어주는 것이다.
-
그러기 위해 d를 오름차순으로 정렬할 필요가 있다.
적은 예산부터 하나씩 더해가면서 지정된 예산과 비교를 해야 한다. 그러므로 요청을 한 부서들이 필요로 하는 예산을 오름차순으로 정렬할 필요가 있다. -
요청한 예산을 더해가다 예산을 넘어가면 i를 반환
계속해서 예산을 더하다가 지정된 예산을 넘어가는 순간 그 부서를 제외하고 이전 부서까지는 지원을 해 줄 수 있다는 얘기가 된다. 그런데 그 값이 지정된 예산을 넘어가게 만든 부서의 인덱스 값과 동일하다.어찌보면 당연한 얘기다. 지원하는 부서를 1부터 시작해서 끝까지 가는 중, n번째 부서에서 초과가 났다면 n - 1개의 부서는 지원이 가능한데, 이는 인덱스의 규칙과 동일하기 때문이다.
-
모든 부서를 다 지원하는 경우는 len(d) 반환
반복문 내에서 return 문에 의해 함수가 종료되지 않았다면, 모든 부서를 지원할 수 있다는 얘기이므로 d의 길이를 반환하면 된다.
다른 사람의 풀이
def solution(d, budget):
d.sort()
while budget < sum(d):
d.pop()
return len(d)요소의 값을 더해나가면서 if 문으로 검사하는 것이 아니라 while문과 pop 메서드를 통해 한 번에 처리할 수 있다.
위 경우 d에 들어있는 요소의 합이 budegt(지정된 예산)을 초과할 경우 마지막 요소(요소들 중 가장 많은 예산을 요청하는 부서)를 제거하고 다시 조건을 검사하는 식으로 진행한다.
