[프로그래머스(LV1)] 예산

문제 설명

S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.

물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.

부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.

• 입출력 예
  

#1
1원, 2원, 3원을 신청한 부서의 물품을 구매해주려면 6원이 필요합니다.
1원, 2원, 5원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.
1원, 3원, 4원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.
1원, 3원, 5원을 신청한 부서의 물품을 구매해주려면 9원이 필요합니다.

3개 부서보다 더 많은 부서의 물품을 구매해 줄 수는 없으므로 최대 3개 부서의 물품을 구매해 줄 수 있습니다.

#2
모든 부서의 물품을 구매해주면 10원이 됩니다. 따라서 최대 4개 부서의 물품을 구매해 줄 수 있습니다.

---

내가 작성한 코드

import Foundation

func solution(_ d:[Int], _ budget:Int) -> Int {
    
    var total: Int = 0
    var count: Int = 0
    var d = d
    
    d = d.sorted(by: <) // 정렬시킴
    for i in d {
        total += i // 하나씩 더해주면서 budget을 초과하는지 확인
        count += 1
        if total > budget {
            total -= i
            count -= 1
            break
        }
    }
    return count
}

예산이 작은 것부터 포함시켜야 가장 많은 부서의 물품을 구매해줄 수 있다.

작은 것부터 더해주며 예산을 넘어가기 전까지의 부서 수를 count해서 리턴해주었다.

주어진 예산보다 작기만하면 되므로 부서를 하나씩 더해주며 확인해주었다.