Water Container Problem Leetcode Python Coding