Maximum Distinct Elements Leetcode Python