二分、划分型DP-书籍印刷 · Copy Books

1、题目描述

书籍复印 · Copy Books https://www.jiuzhang.com/problem/copy-books/

https://www.jiuzhang.com/problem/copy-books/#tag-highlight

给定 n 本书, 第 i 本书的页数为 pages[i]. 现在有 k 个人来复印这些书籍, 而每个人只能复印编号连续的一段的书,

比如一个人可以复印 pages[0], pages[1], pages[2], 但是不可以只复印 pages[0], pages[2], pages[3] 而不复印 pages[1].

所有人复印的速度是一样的, 复印一页需要花费一分钟, 并且所有人同时开始复印. 怎样分配这 k 个人的任务, 使得这 n 本书能够被尽快复印完?

返回完成复印任务最少需要的分钟数.

输入: pages = [3, 2, 4], k = 2
输出: 5
解释: 第一个人复印前两本书, 耗时 5 分钟. 第二个人复印第三本书, 耗时 4 分钟.

2、代码详解

法一：二分+贪心（O(nlog(sum)) 是该问题时间复杂度上的最优解法）

n:书本数目, sum:总页数

class Solution:
    '''
    给定 n 本书, 第 i 本书的页数为 pages[i].
    现在有 k 个人来复印这些书籍, 而每个人只能复印编号连续的一段的书
    怎样分配这 k 个人的任务, 使得这 n 本书能够被尽快复印完?
    返回完成复印任务最少需要的分钟数.
    '''
    def copyBooks(self, pages, k):
        if not pages:
            return 0
        # 答案的范围在 max(pages)~sum(pages) 之间
        # 完成时间的lower bound: max(pages), 理解为一共有len(pages)个人，每个人只复印一本书
        # 完成时间的upper bound: sum(pages), 理解为只有一个人要独自复印所有的书
        start, end = max(pages), sum(pages)
        # 二分：对于 完成时间 进行二分查找，查找最小的 时间, 使得任务可以完成
        # 找到第一个 不多于K个人能在给定的完成时间里完成 复印len(pages)本书 这一任务的 完成时间
        while start + 1 < end:
            mid = (start + end) // 2  # 每次二分到一个时间 time_limit（时限）

            # 如果这个值 <= k，那么意味着大家花的时间可能可以再少一些
            if self.get_least_people(pages, mid) <= k:
                end = mid
            # 如果 > k 则意味着人数不够，需要降低工作量
            else:
                start = mid

        if self.get_least_people(pages, start) <= k:
            return start

        return end

    # 用贪心法从左到右扫描一下 pages，看看需要多少个人来完成抄袭
    # 给定 时限time_limit，求多少人
    def get_least_people(self, pages, time_limit):
        count = 0
        time_cost = 0
        for page in pages:
            if time_cost + page > time_limit:  # 超出time_limit加人，新人的cost需清零算
                count += 1
                time_cost = 0
            time_cost += page  # 累加单人 连续抄书的时间（不能超出time_limit）

        return count + 1

pages = [3, 2, 4]
k = 2
s = Solution()  # 5, 第一个人复印前两本书, 耗时 5 分钟. 第二个人复印第三本书, 耗时 4 分钟
print(s.copyBooks(pages, k))

法二：划分型DP（Java）

算prefix sum的好处是：在求某一段的和时直接相减就能得到结果，pagesj+...+pagesi = prefixSumi - prefixSumj

public class Solution {
    
    public int copyBooks(int[] A, int K) {
        int n = A.length;
        if (n == 0) return 0;
        if (K > n) K = n;
        int[][] f = new int[K+1][n+1]; // K个copier，抄完n本书需要时间
        // Initialize 
        // 事先设置好前缀和，后面减少一次循环
        int[] prefixA = new int[n+1];
        Arrays.fill(prefixA, 0);
        for (int i = 1; i <= n; ++i) prefixA[i] = prefixA[i-1] + A[i-1];
        // 0 个 copier，无穷大时间
        for (int i = 0; i <= n; ++i) f[0][i] = Integer.MAX_VALUE;
        // 0 本书，不管几个copier耗时都是0
        for (int k = 0; k <= K; ++k) f[k][0] = 0;
        //状态转移方程
        //f[k][i] = min (j=0...i) {max{f[k-1][j], A[j]+...+A[i-1]}}
        for (int k = 1; k <= K; ++k) {  // 枚举 k 从 1 到 K
            for (int i = 1; i <= n; ++i) { // 枚举 书 1 到 n 
                f[k][i] = Integer.MAX_VALUE;
                for (int j = 0; j < i; ++j) {
                    // 枚举之前K-1个人抄了j本书的情况，和剩下的书所时间比较取其大，再刷新当前最小记录
                    f[k][i] = Math.min(f[k][i], Math.max(f[k-1][j], prefixA[i] - prefixA[j]));
                }
            }
        }
        return f[K][n];
    }
}

f[k][i]为前k个抄写员最少需要多少时间抄完前i本书