LeetCode 42 Trapping Rain Water

题意：

给出一个代表高度的数组，求出由它组成的形状能储存多少体积的水。（下例中蓝色表示水）

思路：

本题有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;
        }
    };
};