接雨水暴力算法

最新推荐文章于 2022-09-14 14:45:05 发布

踩着七彩祥云的猴子

最新推荐文章于 2022-09-14 14:45:05 发布

阅读量153

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/weixin_36149892/article/details/108979111

版权

leetcode 专栏收录该内容

148 篇文章 0 订阅

订阅专栏

思路

在当前位置, 考虑左右两边的最大高度, 然后取其中的最小值, 最小值减去当前位置的高度就是可以存的水的体积, 一直累加就可以

Code

class Solution {
public:
    int trap(vector<int>& height) {
        if (height.size() <= 2) {
            return 0;
        }
        int n = height.size();
        int result = 0;
        int leftMax = height[0];
        for (int i = 1; i < n - 1; i++) {
            leftMax = max(leftMax, height[i]);
    
            int rightMax = 0;
            for (int j = i; j < n; j++) {
                rightMax = max(rightMax, height[j]);
            }
            result += min(leftMax, rightMax) - height[i];
        }
        return result;
    }
};

优化版本

提前计算出在i位置的时候左边的最大值和右边的最大值, 然后在计算雨水体积的时候可以直接获取, 省去一层for循环.
本质就是空间换时间

class Solution {
public:
    int trap(vector<int>& height) {
        vector<int> rightMaxs;
        vector<int> leftMaxs;
        int n = height.size();
        int result = 0;
        if (n <= 2) {
            return 0;
        }
        int a = height[0];
        int b = height[n-1];
        for (int i = 0; i < n; i++) {
            a = max(height[i], a);
            leftMaxs.push_back(a);
        }

        for (int j = n - 1; j >= 0; j--) {
            b = max(height[j], b);
            rightMaxs.push_back(b);
        }

        reverse(rightMaxs.begin(), rightMaxs.end());
        for (int i = 1; i < n - 1; i++) {
            result += min(leftMaxs[i], rightMaxs[i]) - height[i];
        }
        return result;    
    }
    
};

更快的版本

不适用vector可变数组, 直接使用最基本的数组
注意数组的初始化, 不使用memset操作不能保证数组中的值为0, 可能是垃圾数据

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        int result = 0;
        if (n <= 2) {
            return 0;
        }
        int rightMaxs[n];
        memset(rightMaxs, 0, sizeof(rightMaxs));
        int leftMaxs[n];
        memset(leftMaxs, 0, sizeof(leftMaxs));
        int a = height[0];
        int b = height[n-1];
        for (int i = 0; i < n; i++) {
            a = max(height[i], a);
            leftMaxs[i] = a;
        }

        for (int j = n - 1; j >= 0; j--) {
            b = max(height[j], b);
            rightMaxs[j] = b;
        }

        for (int i = 1; i < n - 1; i++) {
            result += min(leftMaxs[i], rightMaxs[i]) - height[i];
        }
        return result;    
    }
    
};