这道题本来是用dp做的,这几天都在练dp,看了一下,可以用线段树来做,就尝试了一下,虽然不好写,但是还是可以做的- -。
这题的关键在于预处理出每个方格可以向左向右分别扩展多少长度,记录l[i]和r[i]分别表示第i个格子可以向左向右扩展的格子数,那么如果选中了第i个格子,面积就是(l[i]+r[i]+1)*h[i],这样最后线性扫一遍就可以了。
l[i]的求法,最直接的方法就是枚举,从该格子向左枚举,找到第一个比它小的格子,这个格子右边的都是可扩展的。
这样做复杂度是O(n^2)的,对于100000的数据量来说是不可以接受的,所以我们需要优化,用线段树维护每个区间的最小值,我们会发现这个格子事实上就是i号格子左边区间的最靠右边的比它小的格子,这样就可以轻松转化为给定区间求区间中最靠右的比x小的元素的id。
r[i]的求法类似,所以要维护两棵线段树。
最后问题迎刃而解,还要注意的就是需要用到long long类型。。。
我的代码: