数据结构 修剪树木

这是一个经典的贪心算法问题。我们可以使用二分答案的思想来解决。假设我们已知每个工作人员最多可以完成的任务总耗时为x，那么就尽可能让每个工作人员完成更多的任务。

具体来说，从第一棵树开始，依次将每棵树分配给工作人员，直到当前工作人员的任务总耗时大于x，此时需要换下一个工作人员。如果最后所有树木都被分配完了，那么说明当前的x可以满足条件，需要尝试更小的值；否则，当前的x太小，需要尝试更大的值。

该算法的时间复杂度为O(nlogr)，其中r为所有树木的总耗时。

以下是该算法的C++代码实现：

#include <iostream>
using namespace std;

const int N = 100010;

int n, m;
int a[N];

// 判断当前时间限制x是否能够满足要求
bool check(int x)
{
    int cnt = 1, sum = 0;
    for (int i = 1; i <= n; i++)
    {
        // 当前工作人员无法完成当前树木的修剪任务
        if (a[i] > x) return false;
        // 当前工作人员还能够继续完成任务
        if (sum + a[i] <= x) sum += a[i];
        // 需要将当前任务分配给下一个工作人员完成
        else
        {
            cnt++;  // 给下一个工作人员分配任务
            sum = a[i];
        }
    }
    return cnt <= m;
}

int main()
{
    cin >> n >> m;
    int l = 1, r = 0;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        r += a[i];
    }

    // 二分答案
    while (l < r)
    {
        int mid = (l + r) >> 1;
        if (check(mid)) r = mid;
        else l = mid + 1;
    }

    cout << l << endl;
    return 0;
}

在这道题中，贪心算法和二分查找都是解决问题的关键。

贪心算法的作用是确定每个工作人员可以负责的任务范围，使得总耗时最小。具体来说，贪心算法从第一棵树开始，依次将每棵树分配给工作人员，直到当前工作人员的任务总耗时大于x，此时需要换下一个工作人员。如果最后所有树木都被分配完了，那么说明当前的x可以满足条件，需要尝试更小的值；否则，当前的x太小，需要尝试更大的值。通过这种贪心策略，可以尽可能地减少总耗时。

二分查找的作用是确定每个工作人员可以负责的任务耗时的上限，使得总耗时最小。具体来说，我们可以二分答案x，然后根据贪心策略判断当前的x是否能够满足条件。如果能够满足条件，说明当前的x可能是一个可行解，需要尝试更小的值；否则，当前的x太小，需要尝试更大的值。通过这种二分查找的方法，可以快速地找到最小的总耗时。