Leetcode Question 21 Python