BOJ-2343 기타 레슨 Swift

문제

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

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

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

입력

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

출력

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

내가 푼 풀이

접근방법: 이분탐색

- 담을수 있는 블루레이의 길이는 최소 1부터 최대 기타강의의 모든길이를 더한값이 된다.

- 이 범위를 이분탐색으로 최소크기를 구한다.

 

 

주의해야할 점)

- 기타강의는 순서가 바뀌면안되므로 주어진 순서를 지켜야한다.

- 블루레이의 크기가 기타강의 한개의 길이보다 작을 수 있다.

- 기타강의를 빼먹지않고 모두 블루레이에 담아야한다.

 

코드로 구현하면 다음과 같다.

import Foundation

// 입력받기
let input = readLine()!.split(separator: " ").map{Int(String($0))!}
let N = input[0], M = input[1]
var arr = readLine()!.split(separator: " ").map{Int(String($0))!}
var start = 1
var end = arr.reduce(0, +)

// 이분탐색
while start <= end {
    // 이분탐색의 중간값
    let mid = (start + end) / 2
    var sum = 0
    var count = 0
    var left = 0
    var pass = false
    
    // 블루레이에 강의를 최대한 담아본다.
    // 강의길이가 mid값보다 크다면 아에 담을 수 없으므로 이 길이는 사용할 수 없다 (더 길게해서 담아봐야함)
    while left < arr.count {
        sum += arr[left]
        if sum > mid {
            let num = sum - arr[left]
            if num != 0 && num <= mid {
                count += 1
            } else {
                pass = true
                break
            }
            sum = arr[left]
        }
        left += 1
    }
    // 최대한 담아보고 남은 강의가 있고, mid보다 작다면 블루레이에 담을 수 있으므로 +1
    // 남은 강의가 있지만 블루레이보다 크다면 담을 수 없다. pass = true
    if sum != 0 && sum <= mid {
        count += 1
    } else {
        pass = true
    }
    // 블루레이 크기가 너무작아 모든강의를 담을 수 없었다면 블루레이 크기를 늘린다.
    if pass {
        start = mid + 1
    } else {
        // 모든 강의를 담아봤지만 정한 갯수보다 작거나 같다면 갯수를 늘리기위해 블루레이 크기를 줄인다.
        // 모든 강의를 담아봤지만 정한 갯수보다 크다면 갯수를 줄이기위해 블루레이 크기를 늘린다.
        if count <= M {
            end = mid - 1
        } else {
            start = mid + 1
        }
    }
}
print(start)