[Summer/Winter Coding(~2018) 예산]
문제 설명
S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.
물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.
부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.
제한사항
- d는 부서별로 신청한 금액이 들어있는 배열이며, 길이(전체 부서의 개수)는 1 이상 100 이하입니다.
- d의 각 원소는 부서별로 신청한 금액을 나타내며, 부서별 신청 금액은 1 이상 100,000 이하의 자연수입니다.
- budget은 예산을 나타내며, 1 이상 10,000,000 이하의 자연수입니다.
풀이
def solution(d, budget):
answer = 0
curSum = 0
newD = sorted(d)
while (curSum < budget):
## runtimeError를 피하기 위해서 추가한 if문
if (len(d) == answer):
return answer
break
##
curSum += newD[answer]
if (curSum <= budget):
answer += 1
return answer대체적으로 코드는 다들 비슷하게 짠 것 같았다 처음에는 curSum이 budget 보다 작을때 게속 while loop을 돌리는 걸로 생각했는데, 더해진 이후를 계산을 못해주는 것 같아서 마지막 부분에 있는 if 문을 추가했다.
테스트케이스에서는 모두 통과였는데, 막상 실행해보니까 런타임 에러가 나길래 윗줄의 if 문을 하나 더 추가해주었다.
간단하게 만약에 len(d)가 answer, 즉 여태까지 더한 부서의 수와 같다면 그냥 그자체의 길이를 return해서 break 하도록 만들었다. 더이상 loop을 돌리지 않아도 되니까 runTimeError는 사라졌다.
다른 사람 풀이
def solution(d, budget):
d.sort()
while budget < sum(d):
d.pop()
return len(d)아직 잘 이해가 안가는 풀이다.. 왜..?
