青云算法面试题干货-Koko吃香蕉-LeetCode第875题

题目：Koko是一只喜欢吃香蕉的大猩猩。现在有N堆香蕉，第i堆有piles[i]个香蕉。管理员现在离开了并且在H（H >= piles.length）小时后回来。Koko可以决定吃香蕉的速度，即每小时吃K个香蕉。每个小时它选择一堆香蕉，并吃掉其中的K个香蕉。如果某一堆里剩下的香蕉少于K个，它会吃完这一堆剩下的所有香蕉，但它这一小时内不会吃更多的其他堆的香蕉。Koko喜欢慢慢地吃香蕉，但又想在管理员回来之前吃完所有香蕉。请问Koko每小时至少需要吃多少香蕉才能在管理员回来之前吃完所有香蕉。例如有4堆香蕉，每堆香蕉的数目为[3, 6, 7, 11]，如果管理员在8小时后回来，那么Koko每小时吃4个香蕉。

分析：这是LeetCode第875题。

由于Koko在一个小时内把一堆香蕉吃完之后不会再去吃其他的香蕉，那么它一小时能吃掉的香蕉的数目不会超过最多的一堆香蕉的数目（记为M）。同时，它每小时最少会吃1个香蕉。最终Koko决定的吃香蕉的速度K应该是在1到M之间。

我们可以应用二分查找的思路，先选取1和M的平均数，(1+M)/2，看以这个速度Koko能否在H小时内吃掉所有香蕉。如果不能在H小时内吃掉所有的香蕉，那么它需要尝试更快的速度，也就是K应该在(1+M)/2到M之间，下一次我们尝试(1+M)/2和M的平均值。

如果Koko以(1+M)/2的速度能够在H小时内吃完所有的香蕉，那么我们判断这是不是最慢的速度。我蛮尝试一下稍微慢一点的速度，(1+m)/2 - 1。如果Koko以这个速度不能在H小时之内吃完所有香蕉，那么(1+M)/2就是最慢的可以在H小时吃完香蕉的速度。如果以(1+m)/2 - 1的速度也能在H小时内吃完香蕉，那么接下来Koko尝试更慢的速度，1和(1+M)/2的平均值。

以此类推，我们按照二分查找的思路总能找到让Koko在H小时内吃完所有香蕉的最慢速度K。以下是这种思路的参考代码：

public int minEatingSpeed(int[] piles, int H) {
    int max = Integer.MIN_VALUE;
    for (int pile : piles) {
        max = Math.max(max, pile);
    }

    int left = 1;
    int right = max;
    while (left <= right) {
        int mid = left + (right - left) / 2;
        int hours = getHours(piles, mid);
        if (hours <= H) {
            if (mid == left || getHours(piles, mid - 1) > H) {
                return mid;
            }

            right = mid - 1;
        } else {
            left = mid + 1;
        }
    }

    return -1;
}

private int getHours(int[] piles, int speed) {
    int hours = 0;
    for (int pile : piles) {
        hours += (pile + speed - 1) / speed;
    }

    return hours;
}

辅助函数getHours的作用是计算Koko以某一速度吃完所有香蕉需要的时间。

在上述代码中，getHours需要扫描整个数组，每执行一次需要O(n)的时间。函数getHours执行的次数等于minEatingSpeed中while循环执行的次数。由于应用了二分查找算法，while循环执行的次数为O(logMAX)，MAX为最多一堆香蕉的数目。因此总的时间复杂度是O(nlogMAX)。

更多算法面试题的讨论，欢迎访问博客http://qingyun.io/blogs。