Python Leetcode Solution