leetcode#42 Trapping rain water的五种解法详解

时间:2024-08-05 23:34:20

leetcode#42 Trapping rain water

这道题十分有意思,可以用很多方法做出来,每种方法的思想都值得让人细细体会。

42. Trapping Rain Water
Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.

For example,
Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.

leetcode#42 Trapping rain water的五种解法详解

Solution 1:

通过分别计算每一坐标i上有多少水,进而将其相加得到答案。

问题是我们如何知道每一坐标i上有多少水呢?仔细思考,其实只有出现“两高夹一矮”才可能会存到水,如下图所示。

leetcode#42 Trapping rain water的五种解法详解

进而,我们可以想到:每一坐标i上存多少水是由 1.其自身高度 2.它左边的最高高度left_most 3.它右边的最高高度right_most三种因素决定的。

当 min{ left_most, right_most} 小于或等于其自身高度时,它能存的水就是0,比如array[1]=1,其left_most= array[0]=0, 其right_most=array[7]=3, min{left_most, right_most}=left_most=0< height= array[1]=1,这也就是说坐标1 存不了水。

当min{ left_most,right_most} 大于其自身高度时,这时这三者间出现了“两高夹一矮”的情况,故其能存水,而且其存水数= min{left_most,right_most} - height。

我们分别来对一些坐标进行验证:

坐标1,存水数=0.//正确

坐标2,leff_most=1,right_most=3,存水数=left_most-height=1-0=1.//正确

坐标3,left_most=1,right_most=3,min{left_most,right_most}=1=height,存水数=0.//正确

读者可以对每个坐标进行验证,会发现以上结论皆是正确的。所以,现在我们的solution就出来了,我们只需要求出每个坐标对应的left_most和right_most,再把存水数相加,就是总的存水数了。

所以,很朴素自然的一个想法就是,遍历一遍数组,对每个数组元素遍历左边一次求出left_most,遍历右边一次求出right_most。

代码如下,

 
//29ms 6.36%
//complexity: O(N^2)
int trap(vector<int>& height)
{
int ans = 0;
int size = height.size();
for (int i = 1; i < size - 1; i++) {
int max_left = 0, max_right = 0;
for (int j = i; j >= 0; j--) { //Search the left part for max bar size
max_left = max(max_left, height[j]);
}
for (int j = i; j < size; j++) { //Search the right part for max bar size
max_right = max(max_right, height[j]);
}
ans += min(max_left, max_right) - height[i];
}
return ans;
}

Solution 2:

在solution 1里,我们已经知道只要求出left_most和right_most,就可以求出答案,那能不能优化一下求这两个数的过程呢?当然是可以的,我们只需要左遍历一次数组,右遍历一次数组,即可得到left_most和right_most。

/*Solution2: 上一种方法其实有优化的空间
通过两次for循环可分别求得left_most和right_most,第三次for循环即可求得sum,
complexity: O(n)
*/
int trap(vector<int>& height)
{
if(height == null)
return ;
int ans = ;
int size = height.size();
vector<int> left_max(size), right_max(size);
left_max[] = height[];
for (int i = ; i < size; i++) {
left_max[i] = max(height[i], left_max[i - ]);
}
right_max[size - ] = height[size - ];
for (int i = size - ; i >= ; i--) {
right_max[i] = max(height[i], right_max[i + ]);
}
for (int i = ; i < size - ; i++) {
ans += min(left_max[i], right_max[i]) - height[i];
}
return ans;
}

Solution 3:

这里再介绍一种优化方法,双指针法,在数组首尾分别创建一个指针,两指针相见时结束循环。

int trap(vector<int>& height)
{
int left = , right = height.size() - ;
int ans = ;
int left_max = , right_max = ;
while (left < right) {
if (height[left] < height[right]) {
height[left] >= left_max ? (left_max = height[left]) : ans += (left_max - height[left]);
++left;
}
else {
height[right] >= right_max ? (right_max = height[right]) : ans += (right_max - height[right]);
--right;
}
}
return ans;
}

Solution 4:

既然可以纵向的求存水数,那我们能不能一层一层的求存水数呢?

leetcode#42 Trapping rain water的五种解法详解

这是第一层,当我们遇到一个空的,且不在边界,存水数+1,所以第一层我们在i=2,i=5 时分别+1.

leetcode#42 Trapping rain water的五种解法详解

第二层,存水数+4,依次类推,最终可以求出答案。

代码笔者就不给了,读者有兴趣的可以自己写来试试。

Soluton 5:

这是在leetcode中solution给出的一种很新颖的解法,利用了栈的结构,通过维护一个非递增栈来得到答案。

本质思想还是利用了要存水必须是“两高夹一矮”这个特点,只不过这里是用非递增栈来实现。

leetcode#42 Trapping rain water的五种解法详解

下面定义一些符号以便理解:

stack[-1] 栈顶元素

stack[-2] 栈顶的下面一个元素(即倒数第二个元素)

solution4的整个算法是这么实现的:遍历数组,遇到一个元素时,将其与栈顶元素比较,如果其小于等于栈顶元素,直接压栈,将其放入栈中(为维护非递增栈的结构,不能将比栈顶元素大的元素压栈),

若是其大于栈顶元素,此时一定形成了一个“两高夹一矮”局面,因为栈是非递增栈,所以 stack[-1]<stack[-2],又 current>stack[-1],所以是一个“两高夹一矮”局面,此时算完存水数后栈顶元素出栈,继续判断,

递归处理即可。

在上例中整个过程是这样的。

step0: 0不入栈

step1: 1>0 array[1] 入栈 栈:[1]

step2: 0<stack[-1]=1 入栈 栈:[1,0]

step3: 2>stack[-1]=0 存水数+1,0出栈,2>stack[-1]=1, 此时stack内元素不足2,不足以形成“两高夹一矮”局面, 1出栈,2入栈 栈:[2]

step4: 1<stack[-1]=2 1入栈 栈:[2,1]

step5: 0<stack[-1]=1 0入栈 栈:[2,1,0]

step6: 1>stack[-1]=0 存水数+1,0出栈 1=stack[-1] 1入栈 栈:[2,1,1]

step7: 3>stack[-1]=1 存水数+0,1出栈 3>stack[-1]=1 存水数+3,1出栈 3>stack[-1]=2 存水数+0 2出栈 3入栈 栈:[3]

step8: 2<stack[-1] 2入栈 栈:[3,2]

step9: 1<stack[-1] 1入栈 栈:[3,2,1]

step10: 2>stack[-1] 存水数+1 1出栈 2入栈 栈:[3,2,2]

step 11:1<stack[-1] 入栈 栈:[3,2,2,1]

done

/*Solution4
Stack solution
这个solution利用了栈结构,通过维护一个非递增栈,一步一步算出ans
*/ int trap(vector<int>& height)
{
int ans = , current = ;
stack<int> st;
while (current < height.size()) {
while (!st.empty() && height[current] > height[st.top()]) {
int top = st.top();
st.pop();
if (st.empty())
break;
int distance = current - st.top() - ;
int bounded_height = min(height[current], height[st.top()]) - height[top];
ans += distance * bounded_height;
}
st.push(current++);
}
return ans;
}