LeetCode in Python