leetcode记录42. Trapping Rain Water

最新推荐文章于 2023-08-07 09:17:41 发布

原创最新推荐文章于 2023-08-07 09:17:41 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

37 篇文章

订阅专栏

本文详细解析了LeetCode上的一道经典算法题目“Trapping Rain Water”，介绍了如何使用栈来解决这个问题，通过逐行代码解释，展示了如何计算两个高墙间可以储存的雨水量。算法在效率上表现优异，运行时间和内存使用均达到较高水平。

leetcode记录2. Trapping Rain Water

Category	Difficulty	Likes	Dislikes
algorithms	Hard (44.24%)	4173	77

√ Accepted
  √ 315/315 cases passed (4 ms)
  √ Your runtime beats 95.99 % of cpp submissions
  √ Your memory usage beats 8.86 % of cpp submissions (9.5 MB)

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> s;
        int result = 0;
        if (height.size() <= 2)
        {
            return 0;
        }
        s.push(0);
        for (int i = 1; i < height.size(); i++)
        {
            int h = s.top();
            if (height[i] > height[h])
            {
                if(s.size() == 1)
                {
                    s.pop();
                    s.push(i);
                }
                else
                {
                    s.pop();
                    result += min(height[s.top()]-height[h], height[i]-height[h]) * (i-s.top()-1);
                    if (height[s.top()] > height[i])
                    {
                        s.push(i);
                    }                
                    else
                    {
                        i--;
                    }
                    
                }
            }
            else
            {
                s.push(i);
            }
            
        }
        return result;
        
    }
};