Leetcode 1366 Python Tricks