42. Trapping Rain Water

[kazuki-official]

https://leetcode.com/problems/trapping-rain-water/description/

[nodchip]

参考までに少し前に解いたときのコードを貼ります。 3-pass で解いています。座標 i に溜まる水の量は、 min(i より左の一番高い壁の高さ, i より右の一番高い壁の高さ) - i の高さ を求めています。

class Solution {
public:
    int trap(vector<int>& height) {
        vector<int> max_height_from_left(height.begin(), height.end());
        for (int i = 1; i < height.size(); ++i) {
            max_height_from_left[i] = std::max(max_height_from_left[i], max_height_from_left[i - 1]);
        }

        vector<int> max_height_from_right(height.begin(), height.end());
        for (int i = height.size() - 2; i >= 0; --i) {
            max_height_from_right[i] = std::max(max_height_from_right[i], max_height_from_right[i + 1]);
        }

        int volume = 0;
        for (int i = 0; i < height.size(); ++i) {
            volume += std::min(max_height_from_left[i], max_height_from_right[i]) - height[i];
        }

        return volume;
    }
};

[kitano-kazuki]

3 passの方が意図が明確で読みやすいですね。共有ありがとうございます