[백준/C++] 6236번 용돈 관리

dev.hyeon·2022년 2월 10일
0

알고리즘

목록 보기
2/44
post-thumbnail

6236번: 용돈 관리

풀이

이분 탐색을 통해 인출 금액의 최소를 찾는다. 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;
}

0개의 댓글