输入一个整数数组 height
,每个元素相当于一堵墙的高度,两堵墙之间距离为1,要求输出这些墙壁中能存住最多水的两堵墙
思路:
使用贪心法,从左右两端开始,计算最外侧两堵墙所能存住的水体积,并且每次放弃较矮的那堵墙,选择这一侧更靠近中心的下一堵墙继续尝试,直到两堵墙相遇。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
| class Solution(object): def maxArea(self, height): """ :type height: List[int] :rtype: int """ if height is None or len(height) < 2: return 0 maxVol = 0 left, right = 0, len(height) - 1 while left < right: maxVol = max(maxVol, (right - left) * min(height[left], height[right])) if height[left] <= height[right]: left += 1 else: right -= 1 return maxVol
|