Water Container Problem Leetcode Python Basic Programs