Python Leetcode 502