【算法设计作业】week2

来源:互联网 发布:泰国蛇毒洗面奶知乎 编辑:程序博客网 时间:2024/06/13 12:15

照例选题写解析
这里写图片描述

1.Trapping Rain Water

题意

这里写图片描述
大概意思就是给一个数组,代表石头高度,问你一共能装多少水。

思路

1.首先要知道,每一点能装的水 = min(左边最高石头高度-右边最高石头高度)-该点的高度
2.于是问题就变成求每一点左边最高石头高度和右边最高石头高度。一个很直观的思路是,双重循环,暴力解题,复杂度为O(n^2)。但是我觉得不优美,于是不用,但又想不出。

于是看了解析,可以用的办法是动态规划。既要求第i个左边最高的高度,只要用i-1左边最高高度和i高度相比,取最大值就好了。i-1依次类推,于是复杂度就降到了O(n),amazing!

代码

class Solution {public:    int trap(vector<int>& h) {        if (h.size() < 3) return 0;        int ans = 0;        // 这里注意,初始化vector要传入长度,不然后面用[]访问的时候会说引用空指针        vector<int> leftMax(h.size()), rightMax(h.size());        leftMax[0] = h[0];        rightMax[h.size()-1] = h[h.size()-1];        for(int i = 1; i < h.size(); i++) {            leftMax[i] = max(leftMax[i-1], h[i]);        }        for(int i = h.size()-2; i > -1; i--) {            rightMax[i] = max(rightMax[i+1], h[i]);        }        for(int i = 1; i < h.size() - 1; i++) {            ans += min(leftMax[i], rightMax[i]) - h[i];        }        return ans;    }};
原创粉丝点击