[스터디/알고리즘] 이진탐색 백준 2343번
백준 2343번
- https://www.acmicpc.net/problem/2343
문제
강토는 자신의 기타 강의 동영상을 블루레이로 만들어 판매하려고 한다. 블루레이에는 총 N개의 강의가 들어가는데, 블루레이를 녹화할 때, 강의의 순서가 바뀌면 안 된다. 순서가 뒤바뀌는 경우에는 강의의 흐름이 끊겨, 학생들이 대혼란에 빠질 수 있기 때문이다. 즉, i번 강의와 j번 강의를 같은 블루레이에 녹화하려면 i와 j 사이의 모든 강의도 같은 블루레이에 녹화해야 한다.
강토는 이 블루레이가 얼마나 팔릴지 아직 알 수 없기 때문에, 블루레이의 개수를 가급적 줄이려고 한다. 오랜 고민 끝에 강토는 M개의 블루레이에 모든 기타 강의 동영상을 녹화하기로 했다. 이때, 블루레이의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 단, M개의 블루레이는 모두 같은 크기이어야 한다.
강토의 각 강의의 길이가 분 단위(자연수)로 주어진다. 이때, 가능한 블루레이의 크기 중 최소를 구하는 프로그램을 작성하시오.
코드 풀이
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
int n, m;
int arr[100001];
int main() {
ios::sync_with_stdio(false);
cin.tie(NULL);
int l, r = 0;
cin >> n >> m;
for (int i = 0; i < n; i++) {
cin >> arr[i];
r += arr[i]; // 최대값 r은 모든 배열의 합
}
l = *max_element(arr, arr + n); // 최소값 l은 배열의 max값
while (l <= r) {
int mid = (l + r) / 2; // 현재 블루레이 길이
int sum = 0, cnt = 0; // mid일때 블루레이 수 구하기
for (int i = 0; i < n; i++) {
if (sum + arr[i] > mid) {
sum = 0;
cnt++;
}
sum += arr[i];
}
if (sum != 0) cnt++;
// 블루레이 수와 m 비교하여 l,r값 조정하기
if (cnt > m) {
l = mid + 1;
}
else {
r = mid - 1;
}
}
cout << l;
return 0;
}
설명
- 레슨 비디오의 개수와 이를 나눠담을 블루레이의 개수있다. 이후 각 비디오의 길이가 있다.
블루레이로 레슨 비디오를 만들려고 한다. 이 때, 블루레이의 최소값을 구한다. (모든 블루레이 길이는 같음.)
-
이진탐색의 이용
-
끝값 : 모든 강의의 시간을 합한 값 ( = 입력받은 배열의 총 합)
-
시작값 : 모든 강의의 시간중에서 제일 큰 값(모든 강의가 레코드 하나에는 들어가야 한다. = 입력받은 배열의 최대값)
-
기준 : 블루레이의 길이
-
mid에 비교한 값이 더 크면 더 큰 값인 오른쪽을 검색하고 더 작으면 왼쪽 값들을 검색한다.