Leetcode Problem In Python