leetcode 11-20
11.盛最多水的容器
解题思路
首先可以想到一个贪心的思路:对于每个候选的右边界j,它应该匹配的左边界i是最靠左的且满足h[i]>=h[j]的那个位置(当然还有另外一种情况,左侧边界i充当最小值,这个只要把数组反一下再做一次上面的算法,更新答案就好了。)。
两种情况:固定左端点,找右端点的最大值;固定右端点,找左端点的最大值,最后取max
代码
const int N = 3e4 + 10;
class Solution {
public:
int maxArea(vector<int>& height) {
// 固定右端点
memset(q, -1, sizeof q);
int res = -1, len = 0;
for(int i = 0; i < height.size(); ++i){
if(!len || height[i] > q[len - 1].first){
// 如果栈中有值,那么计算当前端点作为右端点的最大值
if(len) res = max(res, min(height[i], q[len - 1].first) * (i - q[len - 1].second));
// 将当前值加入栈顶
q[len++] = {
height[i], i};
}else{
// 找到单调栈中从小到大排列中,第一个大于height[i]的数,计算其面积
int ll = 0, rr = len - 1;
while(ll < rr){
int mid = ll + rr >> 1;
if(q[mid].first >= height[i]) rr = mid;
else ll = mid + 1;
}
res = max(res, min(height[i], q[rr].first) * (i - q[rr].second));
}
}
// 固定左端点,和上面很像
memset(q, -1, sizeof q);
len = 0;
for(int i = height.size() - 1; i >= 0; --i){
if(!len || height[i] > q[len - 1].first){
// 不同处: 计算面积
if(len) res = max(res, min(height[i], q[len - 1].first) * (i - q[len - 1].second));
q[len ++] = {
height[i], i};
}else{
int ll = 0, rr = len - 1;
while(ll < rr){
int mid = ll + rr >> 1;
if(q[mid].first >= height[i]) rr = mid;
else ll = mid + 1;
}
// 不同处: 计算面积
res = max