机试真题分享——二分查找

Posted 2021-04-09 灵洛的人间乐园

This browser does not support music or audio playback. Please play it in WeChat or another browser.

题目描述

珂珂喜欢吃香蕉。这里有 N 堆香蕉，第 i 堆中有 piles[i] 根香蕉。警卫已经离开了，将在 H 小时后回来。

珂珂可以决定她吃香蕉的速度 K （单位：根/小时）。每个小时，她将会选择一堆香蕉，从中吃掉 K 根。如果这堆香蕉少于 K 根，她将吃掉这堆的所有香蕉，然后这一小时内不会再吃更多的香蕉。

珂珂喜欢慢慢吃，但仍然想在警卫回来前吃掉所有的香蕉。

返回她可以在 H 小时内吃掉所有香蕉的最小速度 K（K 为整数）。

示例 1：

输入: piles = [3,6,7,11], H = 8
输出: 4

示例 2：

输入: piles = [30,11,23,4,20], H = 5
输出: 30

示例 3：

输入: piles = [30,11,23,4,20], H = 6
输出: 23

解题思路

使用二分查找实现。
先初始化最小和最大速度left,right。根据题意，最小速度设为1，最大速度是数组中最大值。
中间速度：mid = (left + right) / 2
mid速度是否能在时间H内吃完所有香蕉？
能，right = mid，向左查找；
不能，left = mid + 1，向右查找。
然后继续循环，找到能吃完的最小速度left。

参考代码

public int minEatingSpeed(int[] piles, int H) {
        // 最小速度
        int left = 1;
        // 最大速度
        int right = 1;
        for (int pile : piles) {
            right = Math.max(right, pile);
        }
        // 二分查找最小速度
        while (left < right) {
            int mid = (left + right)  / 2;
            if (canEat(piles, mid, H)) {
                right = mid;
            } else {
                left = mid + 1;
            }
        }

        return left;
    }

    /**
     * 以当前速度，能否在h时间内吃完
     *
     * @param piles
     * @param speed 当前速度
     * @param h 时间
     * @return
     */
    private static boolean canEat(int[] piles, int speed, int h) {
        int time = 0;
        for (int pile : piles) {
            // 向上取整
            time += Math.ceil(pile * 1.0 / speed);
        }

        return time <= h;
    }

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/koko-eating-bananas
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。