题意:
给出一个代表高度的数组,求出由它组成的形状能储存多少体积的水。(下例中蓝色表示水)
思路:
本题有O(nlogn)排序解法和O(n)单调栈解法。
排序解法——从大到小的排序数字,然后按顺序放置到数轴上。因为储水的高度取决于一段区间左右两端最高的高度,所以处理高度的方式是从大到小。对于当前需要安放的数字,如果它在之前安放的最左和最右两个位置之内,则答案中需要减去一部分(相当于它挤出了水…)如果它能扩充区间大小,则答案增加一定体积。
单调栈解法——假设我们维护一个单调递减的栈,那么以当前需要安放的数字为右端点对能储水多少呢?例如,[0, 6, 0, 4, 0, 2, 0, 5],这时5贡献的答案为:与2围成 (2 - 0) * 1 的空间,与4围成 (4 - 2) * 3 的空间,与6围成 (5 - 4) * 5 的空间。
代码:
/**
* 单调队列版本 9ms
*/
class Solution {
public:
int trap(vector<int> &height) {
if (height.size() <= 2) {
return 0;
}
int top = 1;
int *ht = new int[height.size()];
int *id = new int[height.size()];
ht[0] = height[0];
id[0] = 0;
int ans = 0;
for (int i = 1; i < height.size(); ++i) {
if (height[i] == 0) {
continue;
}
int h = 0;
while (top > 0 && ht[top - 1] <= height[i]) {
ans += (min(ht[top - 1], height[i]) - h) * (i - id[top - 1] - 1);
h = min(ht[top - 1], height[i]);
--top;
}
if (top > 0) {
ans += (min(ht[top - 1], height[i]) - h) * (i - id[top - 1] - 1);
}
ht[top] = height[i];
id[top] = i;
++top;
}
return ans;
}
};
/**
* 排序版本 12ms
*/
class Solution {
public:
int trap(vector<int> &height) {
if (height.size() <= 2) {
return 0;
}
cmp *itm = new cmp[height.size()];
for (int i = 0; i < height.size(); ++i) {
itm[i].idx = i;
itm[i].val = height[i];
}
sort(itm, itm + height.size());
int ans = 0, l = itm[0].idx, r = itm[0].idx;
for (int i = 1; i < height.size(); ++i) {
if (itm[i].idx < l) {
ans += (l - itm[i].idx - 1) * itm[i].val;
l = itm[i].idx;
} else if (itm[i].idx > r) {
ans += (itm[i].idx - r - 1) * itm[i].val;
r = itm[i].idx;
} else {
ans -= itm[i].val;
}
}
return ans;
}
private:
struct cmp {
int idx, val;
bool operator<(const cmp f) const {
return val > f.val;
}
};
};