문제 설명
Given n non-negative integers a1, a2, ..., an
, where each represents a point at coordinate (i, ai)
. n vertical lines are drawn such that the two endpoints of the line i is at (i, ai)
and (i, 0)
. Find two lines, which, together with the x-axis forms a container, such that the container contains the most water.
Notice that you may not slant the container.
Note: A leaf is a node with no children.
Example 1:
Input: height = [1,8,6,2,5,4,8,3,7]
Output: 49
Explanation: The above vertical lines are represented by array [1,8,6,2,5,4,8,3,7]. In this case, the max area of water (blue section) the container can contain is 49.
Example 2:
Input: height = [1,1]
Output: 1
Example 3:
Input: height = [4,3,2,1,4]
Output: 16
Example 4:
Input: height = [1,2,1]
Output: 2
Constraints:
- n == height.length
- 2 <= n <= 105
- 0 <= height[i] <= 104
문제풀이
class Solution {
func maxArea(_ height: [Int]) -> Int {
var area = 0
var leftPoint = 0
var rightPoint = height.endIndex - 1
while leftPoint <= rightPoint {
let _height = min(height[leftPoint], height[rightPoint])
let width = rightPoint - leftPoint
area = max(area, width * _height)
if height[leftPoint] <= height[rightPoint] {
leftPoint += 1
} else {
rightPoint -= 1
}
}
return area
}
}
참고 : https://leetcode.com/problems/container-with-most-water/