Python Leetcode Solution 1