Leetcode 239 Python