[파이썬 PYTHON · 자바 JAVA] LeetCode【Container With Most Water】

Container With Most Water - LeetCode

자바 코드

class Solution {
    public int maxArea(int[] height) {
        
        int left = 0;
        int right = height.length-1;
        int ans = -1;

        while(left<right){
            int w = right - left;
            int heightL = height[left];
            int heightR = height[right];
            int h = Math.min(heightL,heightR);

            ans = Math.max(ans,w*h);

            if (heightL<=heightR){
                left++;
            }else{
                right--;
            }
        }
        return ans;
    }
}

파이썬 코드

class Solution(object):
    def maxArea(self, height):
        left = 0
        right = len(height)-1

        ans = -1

        while left<right:
            w = right - left

            heightL = height[left]
            heightR = height[right]
            h = min(heightL,heightR)

            ans = max(ans,h*w)
            if heightL<=heightR:
                left+=1
            else:
                right-=1
        
        return ans

 

의외로 엄청 쉽게 풀린 문제

 

3Sum 문제랑 Search in Rotated Sorted Array 문제 같은 유형을 많이 풀어보니 풀이가 바로 떠올랐다.

 

두 개의 포인터를 지정해서 푸는 문제가 많은 것 같다.

 

left 는 0에서 시작하고 right는 젤 오른쪽에서 시작한다.

 

그리고 현재 위치에서 넓이를 구한뒤 ans에 업데이트를 해주고 left와 right 중 높이가 낮은 기둥을 구한 뒤,

 

낮은쪽을 한칸씩 이동시켜주고 넓이를 또 업데이트해주면 된다.

 

left<right 를 만족할때까지 반복하면 된다.