[프로그래머스] 예산 Python Code

프로그래머스 코딩 테스트 연습문제인 예산을 Python으로 풀어보겠다.

참고로 이 문제는 Go가 없으므로 JavaScript, C++ 만 올릴 예정이다.

---

문제

이번 문제도 매우 쉽게 이해할 수 있다.

주어진 예산 안에서 최대한 많은 팀에게 지원해주어야 하고,

몇 개의 팀에 지원이 되었는지 답으로 return 하면 된다.

 

---

구현 방법

신청한 금액의 list d를 오름차순으로 sorting하고 돌며 budget에서 빼며 count 하고

budget이 0 미만이 되면 count 된 값을 return 하는 방식으로 구현했다.

 

Code

def solution(d, budget):
    answer = 0
    d.sort()
    for cost in d:
        budget -= cost
        if budget<0:
            break
        answer +=1
    return answer

 

이 문제는 다른 사람의 풀이에서도 따로 pythonic 한 코드를 보지 못했다.

 

지금까지 생각으로는 최선의 코드라고 생각한다.