IOS/Swift Algorithm Level 1

Swift. 예산

 문제.

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

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

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

제한사항

  • d는 부서별로 신청한 금액이 들어있는 배열이며, 길이(전체 부서의 개수)는 1 이상 100 이하입니다.
  • d의 각 원소는 부서별로 신청한 금액을 나타내며, 부서별 신청 금액은 1 이상 100,000 이하의 자연수입니다.
  • budget은 예산을 나타내며, 1 이상 10,000,000 이하의 자연수입니다.

 풀이. 

// 나의 풀이
func solution(_ d:[Int], _ budget:Int) -> Int {
    var bud = budget
    let arrayd = d.sorted()
    var count = 0
    for i in arrayd {
        if bud >= i {
            bud -= i
            count += 1
        } 
    }
    return count
}

// 인상 깊은 풀이
func solution(_ d:[Int], _ budget:Int) -> Int {
    var budget = budget

    return d.sorted().filter{
        budget = budget - $0
        return budget >= 0
    }.count
}
  • 음 이번문제는 잠깐의 고민이 있었지만 쉽게 지나갔던것 같다. 문제가 최대한으로 예산을 지급해야 하므로 들어오는 배열을 정렬하여 차례대로 빼주면 원하는 값이 나오게 되었다.
  • 아래는 필터를 이용해서 푸신분이 있었는데 역시... 많이 보고 배운다

'IOS > Swift Algorithm Level 1' 카테고리의 다른 글

Swift. 비밀지도  (0) 2020.11.09
Swift. 직사각형 별찍기  (0) 2020.11.08
Swift. 내적  (0) 2020.11.06
Swift. x만큼 간격이 있는 n개의 숫자  (0) 2020.11.05
Swift. 행렬의 덧셈( feat. zip )  (0) 2020.11.04