[Python] 프로그래머스 예산

https://school.programmers.co.kr/learn/courses/30/lessons/12982?language=python3 

프로그래머스

 

 

 

문제풀이

def solution(d, budget):
    d = sorted(d)
    i = 0
    cnt = 0
    if len(d) == 1:
        if d[0] > budget:
            return 0
        else:            
            return 1
    for char in d:
        if cnt >= budget:
            break
        i += 1
        cnt += char
    if cnt == budget:
        return i
    elif cnt < budget:
        return i
    else:
        return i-1

 

반례 처리를 좀 더 깔끔하게 할 수 있었을거 같은데 안되서 아쉽다.