[카카오 Summer/Winter Coding(~2018)] 예산

jeeyeonLIM commented 3 years ago

문제 설명

S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.
물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.
부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.

제한사항

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

입출력 예

d	budget	result
[1,3,2,5,4]	9	3
[2,2,3,3]	10	4

입출력 예 설명

입출력 예1

각 부서에서 [1원, 3원, 2원, 5원, 4원]만큼의 금액을 신청했습니다. 만약에, 1원, 2원, 4원을 신청한 부서의 물품을 구매해주면 예산 9원에서 7원이 소비되어 2원이 남습니다. 항상 정확히 신청한 금액만큼 지원해 줘야 하므로 남은 2원으로 나머지 부서를 지원해 주지 않습니다. 위 방법 외에 3개 부서를 지원해 줄 방법들은 다음과 같습니다.

1원, 2원, 3원을 신청한 부서의 물품을 구매해주려면 6원이 필요합니다. 1원, 2원, 5원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다. 1원, 3원, 4원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다. 1원, 3원, 5원을 신청한 부서의 물품을 구매해주려면 9원이 필요합니다. 3개 부서보다 더 많은 부서의 물품을 구매해 줄 수는 없으므로 최대 3개 부서의 물품을 구매해 줄 수 있습니다.

입출력 예1

모든 부서의 물품을 구매해주면 10원이 됩니다. 따라서 최대 4개 부서의 물품을 구매해 줄 수 있습니다.

jeeyeonLIM commented 3 years ago

나의 코드

Ver1. `from itertools import combinations` 활용

우선 전체 조합을 봐주기로 했다. 모듈을 활용했고, 최대한 많은 조합이 있을 때를 출력해야 하므로 range(len(d),0,-1)를 이용하여 큰수부터 반복하도록 했다.
그 후 각각의 조합 수에 따른 신청 금액의 min값만을 추출하고, budget과 비교한다. 만약 예산 내 들어오면 바로 출력하는 경우가 최대 경우의 수가 되므로 출력하고, 연산 속도를 위해 break로 for문을 빠져나온다.
```
from itertools import combinations
def solution(d, budget):
for i in range(len(d),0,-1):
    if min([sum(k) for k in list(combinations(d, i))]) <= budget: # 예산 내 들어오는 조합이 있으면
        return i
        break
```
결과는 좋지 못했다. 결과를 보면 대부분 시간 초과로 실패. 따라서 itertools 모듈을 사용하지 않는 방법으로 바꿔야 한다.

Ver2. sorting 후 필요 예산 적은 부서부터 차례 배치

def solution(d, budget):
    d.sort() # 정렬해서 예산 작은 부서부터 배당
    mysum=0  # 부서들의 예산 담을 변수
    answer=0 # 부서 수 

    for i in d: 
        mysum += i 

        # budget 초과하면 이전 answer return하도록.
        if mysum > budget: 
            return answer

        # if조건 만족하지 않을 시 부서 수 +1 추가
        answer += 1 

    return answer # d내 부서를 다 돌았는데도 예산 초과안될때

다른 분 idea 참고했다. sorting 후 앞에서부터 더해가는 아이디어다.

❗ 하지만 작은 예산부터 더하는 것보다 이후 값을 더했을 때 더 나은 방법이 있지 않을까..? 못찾겠지만 반례가 있을 것 같다는 느낌이 들었다...

jeeyeonLIM commented 3 years ago

다른 사람들 풀이

너무 간결한 답이 많아서 허무했다 .
Ver1. 큰 값 제외해가기
전반적인 아이디어는 같은데, 맨 뒤 큰 수부터 제외해가는 방법.. 가변수를 생성할 필요도 없고 while문을 이용한다는 점이 좋았다.
```
def solution(d, budget):
d.sort()
while budget < sum(d):
    print(budget, sum(d))
    d.pop()

return len(d)
```
Ver2. budget 값을 빼가며 변경해주기
sorting해준 후, 작은 예산부터 총 예산에서 제외해간다. if문으로 만약 예산이 -값이 되면(예산을 초과하면) 멈춰주고, 부서 수를 나타내는 cnt를 이용해서 출력해준다!
```
def solution(d, budget):
d.sort()
cnt = 0
for i in d :
    budget -= i
    if budget < 0 :
           break
    cnt += 1
answer = cnt
return answer
```

jeeyeonLIM commented 3 years ago

이쯤에서 느낀점.

많이 풀고 많이 정리해두고 접근방식을 외우고 다시 풀어보고 .... 지금까지 하던 것처럼 하다보면 할 수 있다!!!!

jeeyeonLIM / coding_test