Python Leetcode Solution 21