码迷,mamicode.com
首页 > 移动开发 > 详细

[LeetCode]题解(python):042-Trapping Rain Water

时间:2015-11-03 22:31:34      阅读:229      评论:0      收藏:0      [点我收藏+]

标签:

题目来源:

  https://leetcode.com/problems/trapping-rain-water/


 

题意分析:

  输入一组数组,代表一个宽度为1的高度地图。问,这个地图在雨后可以收集多少水。例如,输入一个数组[0,1,0,2,1,0,1,3,2,1,2,1],返回的是6.如图所示:

技术分享


 

题目思路:

  这道题目虽然说是hard难度的题目,但是其实不是很难。不难发现,水都是从最高那个数起和第二高数之间。那么这题可以分成两部。①找到数组的最大值。②计算最大值左边和右边分别可以收集多少水;计算左边的时候,首先找到左边最靠左的高度,计算它和最高数之间收集的水和第二高往左的和,形式化表示就是f(i) = w(i,j) + f(j),f(i)代表从i往左收集的水,w(i,j)代表i和j之间收集的水。w(i,j)和容易算,就是i到j之间第二高减去原来高度之和;右边计算类似。时间复杂度是(O(n))。


 

代码(python):

  

技术分享
 1 class Solution(object):
 2     def left(self,height,i):
 3         if i == 0:
 4             return 0
 5         n = i - 1;mi = n;maxh = height[mi]
 6         while n >= 0:
 7             if height[n] >= maxh:
 8                 maxh = height[n];mi = n
 9             n -= 1
10         ans = 0;k = mi
11         while k < i:
12             ans += maxh - height[k]
13             k += 1
14         return ans + self.left(height,mi)
15     def right(self,height,i):
16         size = len(height)
17         if i == size - 1:
18             return 0
19         n = i + 1; mi = n;maxh = height[mi]
20         while n < size:
21             if height[n] >= maxh:
22                 maxh = height[n];mi = n
23             n += 1
24         ans = 0;k = mi
25         while k > i:
26             ans += maxh - height[k];k -=1
27         return ans + self.right(height,mi)
28     def trap(self, height):
29         """
30         :type height: List[int]
31         :rtype: int
32         """
33         s = len(height)
34         if s == 0:
35             return 0
36         mh = height[0];m = 0; i = 1
37         while i < s:
38             if height[i] >= mh:
39                 mh = height[i];m = i
40             i += 1
41         return self.left(height,m) + self.right(height,m)
42         
View Code

 


 

转载请注明出处:http://www.cnblogs.com/chruny/p/4934365.html

[LeetCode]题解(python):042-Trapping Rain Water

标签:

原文地址:http://www.cnblogs.com/chruny/p/4934365.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!