Water Container Problem Leetcode Python Interview