solve.swift

import Foundation

func solution(_ d:[Int], _ budget:Int) -> Int {
    var total = 0
    var count = 0
    
    for department in d.sorted() {
        if department + total <= budget {
            count += 1
            total += department
        } else {
            break
        }
    }
    
    return count
}