[Python/LeetCode]코딩테스트 준비를 위한 파이썬 알고리즘 문제 11. Container With Most Water 해설(Medium level)

문제 11. Container With Most Water

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.

Look at the picture, 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.

문제 예시

 

두 축 좌표 i, j가 있을 때, 해당 인덱스의 값중 작은 것을 높이로 하고, 두 인덱스의 차이를 밑변으로 하는 사각형 중 가장 큰 것을 찾는 문제입니다. 

 

def maxArea(self, height: List[int]) -> int:
        #start & end와 너비 초기값 설정
        start = max_s = 0
        end = len(height) - 1
        
        #note의 조건에 맞춰 end>start인 모든 경우 탐색
        while end > start :
            #start와 end 중 height가 더 작은 것(높이)과 end-start(밑변) 길이를 곱함 
            s = (end - start) * min(height[start], height[end])
            #기존 max_s와 비교해서 큰 값을 취함
            max_s = max(max_s, s)
            
            #만약 end에서의 높이가 더 크면 start의 높이가 더 커지지 않는 이상 관여하지않기 때문에 start를 이동
            #반대의 경우 end를 이동하여 밑변의 길이와 높이를 계속 조정
            if height[start] < height[end] :
                start += 1
            
            else :
                end -= 1
                
        #최종적으로 제일 큰 값을 출력
        return max_s