POJ2559 Largest Rectangle in a Histogram 栈的应用

题目链接 POJ2559 

一.思路分析:
    我看到题目的第一想法,就是木桶效应,由短板确定,所以遍历求出数组中长最短的矩形,乘数组长度即为答案。但是思考一下后,发现这种思路是错误的。例如下图:

反例

短板为1,则框出矩形面积为5,但是若仅框出长为10的一个矩形,则面积为10,大于5,可见思路错误。
    仔细分析一下,就可以发现,对于每个矩形i,如果框出的矩形要完全包含这个矩形,就需要找出它左边和右边第一个比它长度短的矩形,用矩形i的长度乘这段区间的长度,即为完全包含矩形i的最大矩形面积。所以,题目就可以表述为,求每一个矩形i左端和右端第一个比它小的矩形,并计算框出矩形的最大面积area[i],枚举0<=i<n的每个矩形,找出最大的area[i],即为答案。
二、解题方案:
    有了上面的思路,目的就明确了:将每个矩形长存入数组,并求数组中每个元素左边和右边第一个比它小的数的下标。我共想出了三种方案,列举如下:
        1.分治(TLE)
        2.利用优先队列和红黑树,即使用STL容器priority_queue和set(TLE)
        3.利用栈(AC)
三、具体介绍:
    解法一:
    我联想到另一道题目:求数组最大和连续子序列。解法一思路和这题类似,对于数组中每个元素,最大值的取值有三种情况:全部在这个元素左边,全部在这个元素右边,包含这个元素。因此利用分治的思想,递归左边求最大值;递归右边求最大值;遍历数组求出这个元素左边和右边第一个比它小的元素,用元素值成区间长度;最大值即为这三个值的最大值。虽然最好情况下时间复杂度为O(nlogn),但是最坏情况时间复杂度为O(n^2),并不比暴力求解好多少,不出意料,TLE(而且从解法二看出,即使O(nlogn)的复杂度仍然TLE)。

解法一关键代码

    解法二:
    由于这一题的思路很明确,所以要做的就是降低时间复杂度。所以我采用了优先队列和set。对于数组中每个元素a[i],需要知道左右第一个比它小的元素,所以只要从最小的元素开始寻找即可。因此我采用了优先队列,O(n)时间构建一个小根堆,每次取最小元只要O(1),维护小根堆性质每次需要O(logn)时间,因此总时间为O(nlogn)。
    而set里存储的是小于等于当前堆根元素的元素下标,初始set里存-1和n(下文解释原因)。因为set中全部是比堆根元素还要小的元素下标,所以每次查询堆根处的元素的下标在set里相邻的两个元素,就是左右第一个比它小的元素,如果左右侧没有比它小的元素,就可以认为左右侧的下标为-1和n。(解释的很乱,下面上图)

比如数组a[4]={1,4,2,3}(为了直观,set画成数组形式,不画成树)


第三行为要求出的数据


set初始情况


堆根1的下标0入set
堆根2的下标3入set
堆根3的下标2入set
堆根4的下标1入set

由此看出,set中第一个比堆根下标小和第一个比堆根下标大的元素就是要求的区间界,可以用upper_bound()函数来查询,时间复杂度也为O(nlogn),最后再遍历一遍数组,求出元素值成区间长度的最大值,因此总时间复杂度为O(nlogn)。做出来这种方法,我本以为肯定可以AC,没想到,TLE。

结构体定义和重载运算符
解法二关键代码

    解法三:
    解法二的TLE,是出乎我的意料的,甚至让我产生了我的思路是否正确的怀疑。但是经过分析,我认为思路是没有问题的,也就是说,存在O(n)的算法,可以求出每个元素左侧和右侧第一个比它小的元素下标。于是,我想到了利用栈。简单介绍一下,题目的原形,是求数组每个元素右侧比它大的第一个元素。做法就是构建一个栈,初始栈为空。从数组最后一个元素开始反向遍历,如果栈空,则右侧没有比它大的元素;或栈顶大于该元素,则右侧比它大的元素就是栈顶。否则一直出栈。(讲不清就上图)

    例如数组a[5]={4,1,5,2,3}

第二行为要求结果

    从数组最后一个元素反向遍历,由于此时栈为空,所以3右边没有比它大的元素,即为-1,然后3入栈。


3入栈

    由于栈顶3大于2,所以2右侧第一个比它大的元素为3,2入栈。


2入栈

    由于5大于栈顶,出栈,直到栈为空,因此5右侧没有比它大的元素,为-1。


2 3 出栈,5入栈

    由于栈顶5大于1,所以1右侧第一个比它大的元素为5,1入栈


1入栈

    由于栈顶1小于4,出栈;栈顶5大于4,所以4右侧第一个比它大的元素为5,4入栈


1出栈,4入栈

    此时遍历完成,算法结束,时间复杂度为O(n)。


上述算法关键代码

    所以,能求右侧第一个比它大的元素,当然就能求第一个比它小的元素,也就能求左侧第一个比它小的元素。分析到这里,问题解决——遍历两次数组,O(n)时间求出每个元素左右第一个比它小的元素下标,再O(n)遍历一遍,求出元素值乘区间长度的最大值,总时间为O(n)。果然,AC。下附代码

#include<iostream>

#include<stack>

#include<vector>

using namespace std;

struct LR

{

int l;int r;//构建结构体数组,l r分别为当前元素左右第一个比它小的元素下标

};

int main()

{

int num;

vector<long long int> res;//多组输入

while(1)

{

int *inbox;

LR *lr;

cin>>num;

if(num==0)

{

break;

}

inbox=new int[num];

lr=new LR[num];

for(int i=0;i<num;i++)

{

cin>>inbox[i];

}

//遍历两遍,求每个元素左右第一个比它小的元素下标

stack<int> s;

for(int i=num-1;i>=0;i--)

{

while(!s.empty()&&inbox[i]<=inbox[s.top()])

{

s.pop();

}

if(s.empty())

{

lr[i].r=num;

}

else

{

lr[i].r=s.top();

}

s.push(i);

}

while(!s.empty())

{

s.pop();

}

for(int i=0;i<num;i++)

{

while(!s.empty()&&inbox[i]<=inbox[s.top()])

{

s.pop();

}

if(s.empty())

{

lr[i].l=-1;

}

else

{

lr[i].l=s.top();

}

s.push(i);

}

//再遍历一遍,求最大值,即为答案

long long int m=0;

for(int i=0;i<num;i++)

{

if(m<(long long int)inbox[i]*(lr[i].r-lr[i].l-1))

{

m=(long long int)inbox[i]*(lr[i].r-lr[i].l-1);

}

}

res.push_back(m);

delete []inbox;

delete []lr;

}

for(int i=0;i<res.size();i++)

{

cout<<res[i]<<endl;

}


AC记录(忽略我的打码)

    小感想:
    1.基础数据结构不见得没法做赛题。在这一题中,栈这种家喻户晓的数据结构,竟然比优先队列和红黑树更加优秀,所以我很希望能多做到这类题,增加我对每种数据结构的理解,而不是仅仅知道很多数据结构,但都浅尝辄止。
    2.不要轻易放弃。其实我昨晚解法二,已经想要放弃了,但是我还是说服自己,再想想,终于,解决了问题。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 196,200评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,526评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,321评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,601评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,446评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,345评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,753评论 3 387
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,405评论 0 255
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,712评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,743评论 2 314
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,529评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,369评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,770评论 3 300
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,026评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,301评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,732评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,927评论 2 336

推荐阅读更多精彩内容

  • 在C语言中,五种基本数据类型存储空间长度的排列顺序是: A)char B)char=int<=float C)ch...
    夏天再来阅读 3,286评论 0 2
  • 《孙子兵法》是经典中的经典,这是毫无疑问的。我看到的这本书,不仅仅有原文,还有解释,还有名家集注。 我最初看到名家...
    深笑567阅读 179评论 0 1
  • 【老妹儿9岁】 “姐,我饿了” 【仙苏深深的沉思了一下】 “要不我去给你煎个蛋?特别好吃” 【老妹儿撇了撇嘴】 “...
    昱头阅读 244评论 0 0
  • 继续十二字的一生! 二便畅通不用仔细解释,都能自己体会的到! 蔬果,水必不可缺!!粗纤维的食物尽量多吃! 二便畅通...
    酵素农耕蔬菜建秋阅读 175评论 0 0
  • 喜欢悠闲地躺在床上,听窗外下雨清脆的声音。当然,这雨是细密的、舒缓的。 置身于外面的路人纷纷撑起他们珍贵的伞,那伞...
    夏默xl阅读 432评论 3 7