Leetcode 399 Python Approach