标签:
LeetCode传送门 https://leetcode.com/problems/trapping-rain-water/
目标:找出积木能容纳的水的“面积”,如图中黑色部分是积木,蓝色为可容纳水的部分
假设:积木宽度均为1
输入:各个积木的高度
输出:所有积木能容纳水的“面积”
1. 逐一求积木的间隔似乎不太容易。特别对于图中3-7积木间的容积,如果可以先求底部(4-6间)的容积,当求解上层(3-7)的容积时,还需要做额外的处理,如减掉底部的高度。
2. 既然如此,可否先求出3-7间整体的容积,再将两积木之间的积木面积(4、5、6)减去,即可得这个区域的容积?
3. 那么问题转换成了,已知一个积木,如何寻找下一个积木,并计算这个两积木间的容积
4. 尝试一,寻找一个至少不比当前积木低的积木,作为下一个积木,如图例子为当前积木3,下一个积木7。那么之间的容积如何计算呢,简单,积木3的高度乘以两积木间的间隔宽度,再减去,积木3与积木7之间的积木(4、5、6)面积。
5. 问题:下次迭代从何开始?上一个例子,可以从积木7继续算起。然而如果找不到相应的积木呢,那么从积木3的下一个积木4开始。
6. 看起来,似乎正确。于是乎,开始写代码了。然而第一次提交就献给了WA(/(ㄒoㄒ)/~~)
7. 错误的情况,如果当前的积木是一个特别高的积木,后继找不到更高的积木,然而实际上之间是可能有容积的!不赘述了,补救思路是:在寻找更高的积木的同时,记录当前找到的最高的积木。如果没有找到更高的积木,使用当前找到的最高的积木作为下一个积木,并计算容积。
迭代的过程
1. 寻找下一个至少不比当前积木低的积木,寻找的同时,记录当前找到的最高的积木高度
2. 如果找到,则计算两积木之间的容积(计算过程见思考过程2)
3. 如果没有找到,则计算当前积木与当前找到的最高的积木之间的容积
4. 当前积木设置为找到的积木(可能是2或3的情况),继续迭代
O(n)
1 #include <iostream> 2 #include <vector> 3 4 using namespace std; 5 6 class Solution { 7 public: 8 int trap(vector<int>& height) { 9 if (height.size() == 0) { 10 return 0; 11 } 12 13 int water = 0; 14 int preIndex = 0; 15 int preHeight = 0; 16 17 // 找到第一个高度非0的bar 18 while (preIndex < height.size() - 1 && height[preIndex] == 0) { 19 ++preIndex; 20 } 21 22 if (preIndex == height.size() - 1) { 23 return 0; 24 } 25 26 // 遍历 27 while (preIndex < height.size()) { 28 preHeight = height[preIndex]; 29 30 // 寻找更高的或相等的bar 31 // 同时记录遍历过的最高的bar 32 // 如果寻找不到更高的或相等的bar时,使用记录值计算 33 int next = preIndex + 1; 34 int minHeight = 0; 35 int minIndex = next; 36 37 while (next < height.size() && height[next] < preHeight) { 38 if (height[next] > minHeight) { 39 minHeight = height[next]; 40 minIndex = next; 41 } 42 ++next; 43 } 44 45 // 如果找到 46 if (next != height.size()) { 47 water += (preHeight * (next - preIndex - 1)); // 计算总面积 48 49 for (int i = preIndex + 1; i < next; ++i) { // 减去中间bar面积 50 water -= height[i]; 51 } 52 53 preIndex = next; // 从找到的bar开始下一次的迭代 54 } else { 55 water += (minHeight * (minIndex - preIndex - 1)); // 计算总面积 56 57 for (int i = preIndex + 1; i < minIndex; ++i) { // 减去中间bar面积 58 water -= height[i]; 59 } 60 61 preIndex = minIndex; // 从次高的bar开始 62 } 63 } 64 65 return water; 66 } 67 }; 68 69 int main(int argc, char const *argv[]) { 70 Solution solution; 71 vector<int> height = {0,1,0,2,1,0,1,3,2,1,2,1}; 72 cout << solution.trap(height) << endl; 73 return 0; 74 }
大年初一!祝大家猴年快乐啦,已经工作的童鞋升职加薪,要实习、找工作的童鞋offer多多~博主今年也要面对找实习、找工作的人生大事儿了。好久没写博客了,默默的哀悼上个学期,自己瞎折腾,弄出的不成熟的东西。希望猴年是激情,奋斗,收获的一年~
ps:有没有对我感兴趣的boss收留~附交友地址一枚https://github.com/zrss
简单粗暴的resume:https://github.com/zrss/ghostblog,近期再修改下
标签:
原文地址:http://www.cnblogs.com/hzhesi/p/5185073.html