Programmers/프로그래머스-예산-python

문제📖

풀이🙏

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

코드💻

시간초과 코드

# programmers, phase1 : 예산, python3
from itertools import combinations

def solution(d, budget):
    if sum(d) == budget:
        return len(d)
    for i in range(len(d), 1, -1):
        for j in list(combinations(d, i)):
            if sum(j) == budget:
                return i

성공 코드

def solution(d, budget):
    d.sort()
    
    while budget < sum(d):
        d.pop()
    
    return len(d)

결과😎

출처 && 깃허브📝

https://programmers.co.kr/learn/courses/30/lessons/12982