Leetcode 560 Python