[C++] 뮤직비디오 🔥

📄문제

지니레코드에서는 불세출의 가수 조영필의 라이브 동영상을 DVD로 만들어 판매하려 한다.
DVD에는 총 N개의 곡이 들어가는데, DVD에 녹화할 때에는 라이브에서의 순서가 그대로 유지되어야 한다. 순서가 바뀌는 것을 우리의 가수 조영필씨가 매우 싫어한다. 즉, 1번 노래와 5번 노래를 같은 DVD에 녹화하기 위해서는 1번과 5번 사이의 모든 노래도 같은 DVD에 녹화해야한다.
지니레코드 입장에서는 이 DVD가 팔릴 것인지 확신할 수 없기 때문에 이 사업에 낭비되는 DVD를 가급적 줄이려고 한다. 고민 끝에 지니레코드는 M개의 DVD에 모든 동영상을 녹화하기로 하였다. 이 때 DVD의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 그리고 M개의 DVD는 모두 같은 크기여야 제조원가가 적게 들기 때문에 꼭 같은 크기로 해야 한다.

 

⬇️ 입력

첫째 줄에 자연수 N(1≤N≤1,000), M(1≤M≤N)이 주어진다. 다음 줄에는 조영필이 라이브에서 부른 순서대로 부른 곡의 길이가 분 단위로(자연수) 주어진다. 부른 곡의 길이는 10,000분을 넘지 않는다고 가정하자.

9 3 1 2 3 4 5 6 7 8 9

 

⬆️ 출력

첫 번째 줄부터 DVD의 최소 용량 크기를 출력하세요.

17

 

📝 풀이

#include <iostream>
#include <vector>

using namespace std;

int getCnt(vector<int> arr, int N, int size)
{
	int sum = 0;
	int cnt = 1; // 시작 DVD의 개수는 1

	for (int i = 0; i < N; ++i)
	{
		if (sum + arr[i] > size) // DVD의 용량을 초과한경우?
		{
			++cnt; // DVD를 일단 채웠기때문에 1 증가
			sum = arr[i]; // 초과한 다음 분은 sum에 넣기
		}
		else // DVD의 용량을 아직 못채운경우
		{
			sum += arr[i]; // 용량을 더하여 채우기
		}
	}

	return cnt;
}

int main()
{
	int N, M;
	int lt = 0, rt = 0, mid, res, max = INT_MIN;

	scanf_s("%d %d", &N, &M);
	vector<int> arr(N);

	for (int i = 0; i < N; ++i)
	{
		scanf_s("%d", &arr[i]);
		rt += arr[i];
		max = max > arr[i] ? max : arr[i];
	}

	// lt의 크기가 rt의 크기보다 커질때까지 반복
	while (lt <= rt)
	{
		// DVD의 크기는 (lt + rt) / 2
		mid = (lt + rt) / 2;

		// mid(현재 DVD 사이즈)로 나누었을때, DVD의 개수
		int cnt = getCnt(arr, N, mid);

		// DVD의 크기는 max 이상이어야하며, DVD의 개수가 M보다 작거나 같다면?
		if (mid >= max && cnt <= M)
		{
			res = mid; // res는 가능한 정답
			rt = mid - 1; // rt를 줄여 최적의 답 찾기를 한번 더 시도
		}
		// DVD의 크기(mid)가 max보다 작거나, cnt가 M(나눔 수)보다 크다면?
		else
		{
			lt = mid + 1; // lt를 올려 DVD의 크기를 늘려 cnt를 M에 맞추도록 시도
		}
	}

	printf("%d", res);
}

• 이분검색 알고리즘을 응용하여 lt, rt, mid를 사용하여 문제를 해결합니다.
• 각 주석에 썼듯이, 여러 번 검사를 통해 최적의 값을 찾아내는 방법으로 해결하였습니다.