Leetcode 21 Python Solution