입출력 예 설명
입출력 예 #1
각 부서에서 [1원, 3원, 2원, 5원, 4원]만큼의 금액을 신청했습니다. 만약에, 1원, 2원, 4원을 신청한 부서의 물품을 구매해주면 예산 9원에서 7원이 소비되어 2원이 남습니다. 항상 정확히 신청한 금액만큼 지원해 줘야 하므로 남은 2원으로 나머지 부서를 지원해 주지 않습니다. 위 방법 외에 3개 부서를 지원해 줄 방법들은 다음과 같습니다.
-1원, 2원, 3원을 신청한 부서의 물품을 구매해주려면 6원이 필요합니다.
-1원, 2원, 5원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.
-1원, 3원, 4원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.
-1원, 3원, 5원을 신청한 부서의 물품을 구매해주려면 9원이 필요합니다.
3개 부서보다 더 많은 부서의 물품을 구매해 줄 수는 없으므로 최대 3개 부서의 물품을 구매해 줄 수 있습니다.
입출력 예 #2
모든 부서의 물품을 구매해주면 10원이 됩니다. 따라서 최대 4개 부서의 물품을 구매해 줄 수 있습니다.
접근법
먼저 입력받은 부서 별 신청한 금액 리스트를 오름차순으로 정렬시킨 뒤, 하나씩 더해가며 예산을 벗어나는 순간 예산을 지원받은 부서의 수를 반환해 주는 방식으로 접근해 보겠습니다.
나의 코드
def solution(d, budget):
d.sort()
total_buy = []
for i in d:
total_buy.append(i)
if sum(total_buy) > budget:
total_buy = total_buy[:-1]
break
return len(total_buy)
