LeetCode875、爱吃香蕉的珂珂---二分法

题目描述

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

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

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

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

示例 1：

输入：piles = [3,6,7,11], h = 8

输出：4

代码

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#include <sstream>


using namespace std;

class Solution {
private:
    // 计算速度 k 对应的小时数的函数，算出 hour_used
    int calHourUsed(vector<int>& piles, int k) {
        int hourUsed = 0;
        for (int p : piles) {
            hourUsed += ceil((double)p / k);
        }
        return hourUsed;
    }

public:
    int minEatingSpeed(vector<int>& piles, int h) {
        int left = 1, right = *max_element(piles.begin(), piles.end()) + 1;
        while (left < right) {
            int mid = left + (right - left) / 2;
            // 花费时间太少，速度偏大，速度还可以减小，
            // 搜索区间向左折半，right 可以向左移动
            if (calHourUsed(piles, mid) <= h) {
                right = mid;
            }
            else {
                left = mid + 1;
            }
        }
        return left;
    }
};
//piles = [3,6,7,11], h = 8
int main() {
    string input;
    getline(cin, input);
    input = input.substr(1, input.length() - 2);
    stringstream ss(input);
    vector<int> piles;
    string num;

    while (getline(ss, num, ',')) {
        piles.push_back(stoi(num));
    }

    int h;
    cin >> h;

    Solution a;
    cout << a.minEatingSpeed(piles, h);
}

复杂度分析

时间复杂度：O(nlogm)，其中 n 是数组 piles 的长度，m 是数组 piles 中的最大值。需要 O(n) 的时间遍历数组找到最大值 m，二分查找需要执行 O(logm) 轮，每一轮二分查找需要 O(n) 的时间，因此总时间复杂度是 O(nlogm)。

空间复杂度：O(1)。