Python Leetcode 1008