28-Dolchae

[Dolchae]

🔗 문제 링크

6236 용돈 관리

✔️ 소요된 시간

50분 + a

✨ 수도 코드

문제

• N일 동안 M번만 통장에서 돈을 빼서 쓴다.
• 통장에서 K원을 인출하고 이 돈으로 하루를 보낼 수 있다면 그대로 사용하고, 모자라면 남은 금액은 통장에 넣고 다시 K원을 인출한다.
• 정확히 M번을 맞추기 위해 남은 금액이 사용할 금액보다 많아도 다시 집어넣고 K원을 인출할 수 있다.
• 여기서 최소 금액 K를 구하는 문제이다.

이분 탐색

• 이분 탐색이란 정렬된 리스트에서 특정한 값의 위치를 찾는 알고리즘이다.
• 일반탐색보다 속도가 빠르다.
  • 처음과 마지막의 중간값을 선택해 최종 찾아야하는 값과 크고 작음을 비교하는 방식을 반복하며 탐색한다.

코드

import sys

input = sys.stdin.readline

n, m = map(int, input().split())

money = []
for _ in range(n):
    money.append(int(input()))
start = max(money)  
end = sum(money)  

• n과 m을 입력받고 money에 각각의 날에 사용할 금액을 입력받아 리스트로 저장한다.
• 이분 탐색을 하기 위해 start와 end 변수를 만든다.
  • start는 가장 작은 범위를 나타내는데, 적어도 돈의 최댓값은 쓸 것이므로 max(money)로 지정한다.
  • end는 가장 큰 범위 끝을 나타내며 한번에 인출해 다 쓰는 경우가 이를 나타낸다.

ans = 0
while start <= end:
    mid = (start + end) // 2

    total = 0
    cnt = 0
    for x in money:
        if total + x <= mid:
            total += x
        else:
            cnt += 1
            total = 0
            total += x
    if total:  # 마지막 돈에 대해서 total에 금액이 있으면 cnt++ 처리.
        cnt += 1

• 후에 출력할 최소 k 값인 ans를 0으로 선언해준다.
• while문을 반복하며 이분 탐색을 진행한다.
  • mid는 탐색 중인 범위의 중간값을, total은 현재까지의 총 사용 금액을, cnt는 현재까지의 인출 횟수를 저장한다.
  • money 리스트를 순회하면서 현재 total에 남은 금액을 더했을 때 mid(인출 금액)보다 작거나 같다면 현재 날의 사용 금액을 total에 더해준다.
  • 그렇지 않다면(mid 보다 큰 경우) 인출 횟수를 증가시키고 toatl을 다시 0으로 초기화한 후 현재 날의 사용 금액을 total에 더한다.
• for문이 끝난 후에 total에 남은 금액이 있다면 인출 횟수를 1 더해준다.

while start <= end:
    ```

    if cnt > m:  
        start = mid + 1
    else:  
        end = mid - 1
        ans = mid

print(ans)

• 인출 횟수 cnt가 m보다 크다면 인출 금액을 늘려야 하므로 start를 mid+1로 설정한다.
• 그렇지 않다면 인출 금액을 줄여야 하므로 end를 mid-1로 설정하고 현재 mid를 정답으로 저장한다.

전체 코드

import sys

input = sys.stdin.readline

n, m = map(int, input().split())

money = []
for _ in range(n):
    money.append(int(input()))

start = max(money)
end = sum(money)

ans = 0
while start <= end:
    mid = (start + end) // 2

    total = 0
    cnt = 0
    for x in money:
        if total + x <= mid:
            total += x
        else:
            cnt += 1
            total = 0
            total += x
    if total:  
        cnt += 1

    if cnt > m:  
        start = mid + 1
    else:  
        end = mid - 1
        ans = mid

print(ans)

📚 새롭게 알게된 내용

이분 탐색을 공부하였다. 개념 자체는 그렇게 어렵게 느껴지지는 않았는데 역시나 문제를 풀려고 하니 많이 어려웠다🥲😂


학습 참고 블로그
코드 참고 블로그

[9kyo-hwang]

이진 탐색... 참 적응 안되는 유형이죠
저도 여전히 쩔쩔맵니다...

그래도 열심히 노력해봅시다!

[9kyo-hwang]

이진 탐색, 정확히는 매개변수 탐색의 경우 어려운 문제일 수록 start, end 값을 변경시키기 위한 조건 판별 코드가 많이 길어집니다.
이 문제에서는 여기 부분이 되겠죠?

def getTakeoutCnt(mid):
    total, take_out = 1, 0 
    for used in used_money:
        if total + used > mid:
            take_out += 1
            total = used
        else:
            total += used
        
    return take_out

보통은 check() 같은 이름을 많이 사용하는데, 여기는 인출하는 횟수를 구하는 거니까 이렇게 이름 지어봤습니다.

while lo <= hi:
    mid = (lo + hi) // 2

    if getTakeoutCnt(mid) > M:
        lo = mid + 1
    else:
        hi = mid - 1

그럼 이진탐색 파트는 이렇게 심플해지죠?

[9kyo-hwang]

둘 다 27차시로 적혀있네요 ㅋㅋ

[xxubin04]

저도 이번 문제는 이진 탐색 문제였는데 많이 헤맸었네요😂
같이 이진 탐색을 한번 정복해봅시다😁💪🏻
수고 많으셨습니다!!

[gjsk132]

전 end를 모든 경우에서 나올 수 있는 큰 값으로 설정해줬는데, sum(money)로 하는 방법도 있네요!

[gjsk132]

마지막에 결국 start와 end가 같아졌을 때, 해당 값이 정답이 되고, start와 end가 역전됩니다.

여기서, 해당 문제의 경우에는 end가 mid-1이 되면서 자연스럽게 start에 있는 (마지막 단계에서) 변하지 않은 값이 정답이 되게 됩니다.

그래서 ans를 따로 정의하지 않고 print(start)로 바로 출력해줘도 정답이 될 수 있습니다.

start = max(money)
end = sum(money)

while start <= end:
    mid = (start + end) // 2

    total = 0
    cnt = 0
    for x in money:
        if total + x <= mid:
            total += x
        else:
            cnt += 1
            total = 0
            total += x
    if total:  
        cnt += 1

    if cnt > m:  
        start = mid + 1
    else:  
        end = mid - 1

print(start)