Leetcode Solution Python