15.3 Maximum capacity problem¶
Question
Input an array \(ht\), where each element represents the height of a vertical partition. Any two partitions in the array, along with the space between them, can form a container.
The capacity of the container is the product of the height and the width (area), where the height is determined by the shorter partition, and the width is the difference in array indices between the two partitions.
Please select two partitions in the array that maximize the container's capacity and return this maximum capacity. An example is shown in Figure 15-7.
Figure 15-7 Example data for the maximum capacity problem
The container is formed by any two partitions, therefore the state of this problem is represented by the indices of the two partitions, denoted as \([i, j]\).
According to the problem statement, the capacity equals the product of height and width, where the height is determined by the shorter partition, and the width is the difference in array indices between the two partitions. The formula for capacity \(cap[i, j]\) is:
Assuming the length of the array is \(n\), the number of combinations of two partitions (total number of states) is \(C_n^2 = \frac{n(n - 1)}{2}\). The most straightforward approach is to enumerate all possible states, resulting in a time complexity of \(O(n^2)\).
1. Determination of a greedy strategy¶
There is a more efficient solution to this problem. As shown in Figure 15-8, we select a state \([i, j]\) where the indices \(i < j\) and the height \(ht[i] < ht[j]\), meaning \(i\) is the shorter partition, and \(j\) is the taller one.
Figure 15-8 Initial state
As shown in Figure 15-9, if we move the taller partition \(j\) closer to the shorter partition \(i\), the capacity will definitely decrease.
This is because when moving the taller partition \(j\), the width \(j-i\) definitely decreases; and since the height is determined by the shorter partition, the height can only remain the same (if \(i\) remains the shorter partition) or decrease (if the moved \(j\) becomes the shorter partition).
Figure 15-9 State after moving the taller partition inward
Conversely, we can only possibly increase the capacity by moving the shorter partition \(i\) inward. Although the width will definitely decrease, the height may increase (if the moved shorter partition \(i\) becomes taller). For example, in Figure 15-10, the area increases after moving the shorter partition.
Figure 15-10 State after moving the shorter partition inward
This leads us to the greedy strategy for this problem: initialize two pointers at the ends of the container, and in each round, move the pointer corresponding to the shorter partition inward until the two pointers meet.
Figure 15-11 illustrate the execution of the greedy strategy.
- Initially, the pointers \(i\) and \(j\) are positioned at the ends of the array.
- Calculate the current state's capacity \(cap[i, j]\) and update the maximum capacity.
- Compare the heights of partitions \(i\) and \(j\), and move the shorter partition inward by one step.
- Repeat steps
2.and3.until \(i\) and \(j\) meet.
Figure 15-11 The greedy process for maximum capacity problem
2. Implementation¶
The code loops at most \(n\) times, thus the time complexity is \(O(n)\).
The variables \(i\), \(j\), and \(res\) use a constant amount of extra space, thus the space complexity is \(O(1)\).
def max_capacity(ht: list[int]) -> int:
"""Maximum capacity: Greedy"""
# Initialize i, j, making them split the array at both ends
i, j = 0, len(ht) - 1
# Initial maximum capacity is 0
res = 0
# Loop for greedy selection until the two boards meet
while i < j:
# Update maximum capacity
cap = min(ht[i], ht[j]) * (j - i)
res = max(res, cap)
# Move the shorter board inward
if ht[i] < ht[j]:
i += 1
else:
j -= 1
return res
/* Maximum capacity: Greedy */
int maxCapacity(vector<int> &ht) {
// Initialize i, j, making them split the array at both ends
int i = 0, j = ht.size() - 1;
// Initial maximum capacity is 0
int res = 0;
// Loop for greedy selection until the two boards meet
while (i < j) {
// Update maximum capacity
int cap = min(ht[i], ht[j]) * (j - i);
res = max(res, cap);
// Move the shorter board inward
if (ht[i] < ht[j]) {
i++;
} else {
j--;
}
}
return res;
}
/* Maximum capacity: Greedy */
int maxCapacity(int[] ht) {
// Initialize i, j, making them split the array at both ends
int i = 0, j = ht.length - 1;
// Initial maximum capacity is 0
int res = 0;
// Loop for greedy selection until the two boards meet
while (i < j) {
// Update maximum capacity
int cap = Math.min(ht[i], ht[j]) * (j - i);
res = Math.max(res, cap);
// Move the shorter board inward
if (ht[i] < ht[j]) {
i++;
} else {
j--;
}
}
return res;
}
3. Proof of correctness¶
The reason why the greedy method is faster than enumeration is that each round of greedy selection "skips" some states.
For example, under the state \(cap[i, j]\) where \(i\) is the shorter partition and \(j\) is the taller partition, greedily moving the shorter partition \(i\) inward by one step leads to the "skipped" states shown in Figure 15-12. This means that these states' capacities cannot be verified later.
Figure 15-12 States skipped by moving the shorter partition
It is observed that these skipped states are actually all states where the taller partition \(j\) is moved inward. We have already proven that moving the taller partition inward will definitely decrease the capacity. Therefore, the skipped states cannot possibly be the optimal solution, and skipping them does not lead to missing the optimal solution.
The analysis shows that the operation of moving the shorter partition is "safe", and the greedy strategy is effective.