Leetcode 643 Python