LeetCode - Container With Most Water

 

 

Container With Most Water - LeetCode

---

문제

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 line i is at (i, ai) and (i, 0). Find two lines, which together with x-axis forms a container, such that the container contains the most water.

Note: You may not slant the container and n is at least 2.

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:
Input: [1,8,6,2,5,4,8,3,7]
Output: 49

---

일단 brute force 로 풀어본다.

class Solution {
    func maxArea(_ height: [Int]) -> Int {
        let len = height.count
        var answer = 0
        for i in 0..<len {
            for j in i+1..<len {
                var volumn = min(height[i],height[j]) * (j-i)
                if answer < volumn {answer = volumn}
            }
        }
        return answer
    }
}

 

Time Limit Exceed 뜨니까 최적화 시켜본다.

이렇게 범위 줄여나가는 거는 two pointer 로 head 랑 tail 해서 막 비교하던데

1트

class Solution {
    func maxArea(_ height: [Int]) -> Int {
        var head = 0 
        var tail = height.count-1
        var answer = 0
        // if height.count == 2 {
        //     return min(height[0],height[1])
        // }
        while head < tail {
            var minHeight = min(height[head], height[tail])
            let volumn = minHeight * (tail - head)
            print(tail - head , minHeight, volumn)
            if answer < volumn {
                answer = volumn
            }else{
                break
            }
            //만약 1보다 height 크면 head 를 움직일 수 있다. (같거나 작으면 움직일 필요가 엄슴다)
            if height[head] < height[head+1] {
                head += 1
            }
            if height[tail] < height[tail-1] {
                tail += -1
            }
        }
        return answer
    }
}

오케이 안되고

슬쩍 힌트보니 최적 시간이 O(n)인거 같다.

일단 한번을 돌아야 되는 것 같다. 

max 를 찾는 도중에 break로 탈출 할 수는 없고 한바퀴 다 돌면서 찾아야 되는것 같다.

그러면 그냥 left 랑 right 랑 한칸씩 비교하면서 중간에서 만나면 되지 않을까

class Solution {
    func maxArea(_ height: [Int]) -> Int {
        var left = 0 
        var right = height.count - 1
        var answer = 0 
        while left < right {
            let v = min(height[left],height[right]) * (right - left)
            if v > answer {
                answer = v
            }
            if height[left] < height[right] {
                left += 1
            }else {
                right -= 1
            }
        }
        return answer
    }
}

 

left 와 right 높이값을 비교해 낮은 값을 바꿔주는 방식이다. 어쨋건 left 와 right가 만날 때 까지 돌기 때문에 width 는 신경쓸 필요 없다.