[백준 알고리즘][C++] 이분 탐색(binary search) - 2343번 기타 레슨

안녕하세요~

​

이번시간에는 2343번 기타 레슨 문제를 풀어보려고 합니다.

 

www.acmicpc.net/problem/2343

2343번: 기타 레슨

 

명심하셔야 될 것은 일단 문제를 풀기 전에 코딩을 먼저 시작하지 마시고, 반드시 연습장으로 생각을 정리하고! 이 후에 생각이 맞는 것 같다고 확신이 되면 그 때 코딩을 시작하세요. 코드를 작성하면서 문제를 풀게되면 정리가 안된 상태기 때문에 코드 수정이 잦아지고, 정신이 없어지게 됩니다ㅠㅠ

그러니 반드시 노트에 어떻게 풀 것인지에 대한 정리를 다 하시고 코딩을 시작하시기 바랍니다!!

​

binary search 공부에 좋은 문제입니다.

문제

강토는 자신의 기타 레슨 동영상을 블루레이로 만들어 판매하려고 한다. 블루레이에는 총 N개의 레슨이 들어가는데, 블루레이를 녹화할 때, 레슨의 순서가 바뀌면 안 된다. 순서가 뒤바뀌는 경우에는 레슨의 흐름이 끊겨, 학생들이 대혼란에 빠질 수 있기 때문이다. 즉, i번 레슨과 j번 레슨을 같은 블루레이에 녹화하려면 i와 j 사이의 모든 레슨도 같은 블루레이에 녹화해야 한다.

강토는 이 블루레이가 얼마나 팔릴지 아직 알 수 없기 때문에, 블루레이의 개수를 가급적 줄이려고 한다. 오랜 고민 끝에 강토는 M개의 블루레이에 모든 기타 레슨 동영상을 녹화하기로 했다. 이때, 블루레이의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 단, M개의 블루레이는 모두 같은 크기이어야 한다.

강토의 각 레슨의 길이가 분 단위(자연수)로 주어진다. 이때, 가능한 블루레이의 크기 중 최소를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 레슨의 수 N (1 ≤ N ≤ 100,000)과 M (1 ≤ M ≤ N)이 주어진다. 다음 줄에는 강토의 기타 레슨의 길이가 레슨 순서대로 분 단위로(자연수)로 주어진다. 각 레슨의 길이는 10,000분을 넘지 않는다.

출력

첫째 줄에 가능한 블루레이 크기중 최소를 출력한다.

 

예제 입력 1

9 3
1 2 3 4 5 6 7 8 9

예제 출력 1

17

정답 코드

#include <iostream>
using namespace std;
int lession[100001];
int N, M;
int maxVal = -1;
int sum = 0;
void solve()
{
	long long start = maxVal; //최소 값
	long long end = sum; //최대 값
	while (start <= end)
	{
		long long mid = (start + end) / 2;
		int count = 0;
		long long blueRaySize = 0;
		for (int i = 0; i < N; i++)
		{
			//블루레이 크기와 lession i 번째의 합이 현재 threshold보다 크면 카운트를 늘림
			if (blueRaySize + lession[i] > mid)
			{
				count++;
				blueRaySize = 0;
				//넘어가면 현재 값 이전까지 나눠지는 거고 현재 값으로 부터 시작
			}
			blueRaySize = blueRaySize + lession[i];
		}
		count++;

		if (count <= M)
		{
			end = mid - 1;
		}
		else
		{
			start = mid + 1;
		}
	}
	cout << start << endl;
}
int main()
{
	cin >> N >> M;
	for (int i = 0; i < N; i++)
	{
		cin >> lession[i];
		sum += lession[i];
		maxVal = max(maxVal, lession[i]);
	}
	solve();
}