Count Largest Group Leetcode Python