Leetcode 399 Python Tricks