이분 탐색을 통해 인출 금액의 최소를 찾는다. N일동안 M번 인출할 경우 최소 인출 금액은 일별 용돈 중 최대값이고, 1번만 인출할 경우 최소 인출 금액은 N일 동안 사용할 용돈의 총합이다. 따라서 이분 탐색 범위는 (일별 용돈 중 최대값)~(사용할 용돈의 총합)이다.
인출 금액을 mid, 사용하고 남은 용돈을 temp, 인출 횟수를 count라고 하자. i번째에 쓰고 남은 용돈을 계산한다. 이때 남은 용돈이 오늘 사용해야 할 금액보다 작으면 남은 용돈을 넣고 인출을 하므로 count는 1 증가하고 남은 용돈을 인출 금액으로 초기화된다.
총 인출 횟수가 M번 이하이면 인출 금액을 줄여야하므로 right를 옮겨 mid 왼쪽 범위를 탐색한다. 반대로 M번 초과이면 인출 금액을 늘려야 하므로 left를 옮겨 오른쪽 범위를 탐색한다.
#include <iostream>
#include <algorithm>
using namespace std;
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
int money[100001]={0,}, N=0, M=0;
int sumMoney = 0;
cin >> N >> M;
for (int i=0; i < N; i++)
{
cin >> money[i];
sumMoney += money[i];
}
int left = *max_element(money, money + N), right = sumMoney, mid = 0;
while (left <= right)
{
mid = (left + right) / 2;
int temp = mid, count = 1;
for (int i = 0; i < N; i++)
{
if (temp < money[i])
{
count++;
temp = mid;
}
temp -= money[i];
}
if (count <= M)
{
right = mid - 1;
}
else
{
left = mid + 1;
}
}
cout << mid;
return 0;
}