Python Leetcode 1004