照例选题写解析
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;
}
};