Leetcode 399 Python Solution