给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
以上是柱状图的示例,其中每个柱子的宽度为 1,给定的高度为 [2,1,5,6,2,3]。
图中阴影部分为所能勾勒出的最大矩形面积,其面积为 10 个单位。
示例:
输入: [2,1,5,6,2,3]
输出: 10
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/largest-rectangle-in-histogram
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
//动态规划
if(heights.empty()){
return 0;
}
//这是一个单调递增的栈,即从下到上都是递增的,栈顶元素最大,
int res=0;
heights.push_back(0);
int len = heights.size();
stack<int>mystack;
for(int i=0;i<len;i++){
while(!mystack.empty()&&heights[i]<heights[mystack.top()]){
int num = mystack.top();
mystack.pop();
int h = heights[num];
//int left = mystack.top();//这里不能这么写,原因是这里的栈可能是空,这样就会出现问题
res = max(res,h*(mystack.empty()?i:(i-mystack.top()-1)));
}
mystack.push(i);
}
return res;
}
};
数帽子问题也一样:
现有一条排好的队伍,从队首到队尾,队员们都戴着帽子,身高是无序的。假设每个人能看到队伍中在他前面的比他个子矮的人的帽子,(如果出现一个比这个人个子高的人挡住视线,那么此人不能看到高个子前面的任何人的帽子。)现在请计算出这个队伍中一共可以看到多少个帽子?例如给定数组为:[2,1,5,6,2,3](顺序为从队尾到队首)。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
//动态规划
if(heights.empty()){
return 0;
}
//这是一个单调递增的栈,即从下到上都是递增的,栈顶元素最大,
int res=0;
heights.push_back(INT_MAX);
int len = heights.size();
stack<int>mystack;
for(int i=0;i<len;i++){
while(!mystack.empty()&&heights[i]>heights[mystack.top()]){
int num = mystack.top();
mystack.pop();
int h = heights[num];
//int left = mystack.top();//这里不能这么写,原因是这里的栈可能是空,这样就会出现问题
// res = max(res,h*(mystack.empty()?i:(i-mystack.top()-1)));
res += i-num-1;
}
mystack.push(i);
}
return res;
}
};
给一个数组,返回一个大小相同的数组。返回的数组的第i个位置的值应当是,对于原数组中的第i个元素,至少往右走多少步,才能遇到一个比自己大的元素(如果之后没有比自己大的元素,或者已经是最后一个元素,则在返回数组的对应位置放上-1)。
例如给定数组为:[2,1,5,6,2,3]
返回数组应该为:[2,1,1,-1,1,-1]
其实这个问题本质上和数帽子问题是一样的,本质都是找到元素右边第一个比它大的数,代码稍作改动即可。
作者:edward_wang
链接:https://leetcode-cn.com/problems/largest-rectangle-in-histogram/solution/xiang-xi-jie-shao-dan-diao-zhan-de-li-jie-he-shi-y/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。