LeetCode Koko Eating Bananas Solution

Koko loves to eat bananas. There are n piles of bananas, the ith pile has piles[i] bananas. The guards have gone and will come back in h hours.

Koko can decide her bananas-per-hour eating speed of k. Each hour, she chooses some pile of bananas and eats k bananas from that pile. If the pile has less than k bananas, she eats all of them instead and will not eat any more bananas during this hour.

Koko likes to eat slowly but still wants to finish eating all the bananas before the guards return.

Return the minimum integer k such that she can eat all the bananas within h hours.

Example 1:
Input: piles = [3,6,7,11], h = 8
Output: 4

Example 2:
Input: piles = [30,11,23,4,20], h = 5
Output: 30

Example 3:
Input: piles = [30,11,23,4,20], h = 6
Output: 23

Constraints:

  • 1 <= piles.length <= 104
  • piles.length <= h <= 109
  • 1 <= piles[i] <= 109

Problem Link: https://leetcode.com/problems/koko-eating-bananas/



Solution

We want to apply the binary search template, and implement the feasible function. Here, the feasible function is whether Koko can_finish_eating all piles within h hours while eating at speed k per hour.

Since Koko eats at only one pile during each hour, ceil(float(p)/k) is the time Koko takes to finish pile p. Note that p/k does not work here because we want a whole number of hours so we needed to round up p/k. Therefore, the feasiblity is determined by whether Koko's hours_used is within h hours, where hours_used is the total hours to finish all piles.

Recall example 1 with Input: piles = [3,6,7,11], h = 8, and Output: 4, the can_finish_eating function returns the following results.

Implementation

1def can_finish_eating(piles, h, k):
2    hours_used = 0
3    for p in piles:
4        hours_used += ceil(float(p)/k)
5    return hours_used <= h
6
7def minEatingSpeed(piles, h):
8    left, right = 1, 1000000000 # 10^9 max length of the piles
9    ans = -1
10    while left <= right:
11      mid = (left + right) // 2
12      if can_finish_eating(piles, h, mid):
13          ans = mid
14          right = mid - 1
15      else:
16          left = mid + 1
17    return ans