LeetCode-Python-1399. 统计最大组的数目（模拟法 + 哈希表）

 

1399. 统计最大组的数目

难度简单4

给你一个整数 n 。请你先求出从 1 到 n 的每个整数 10 进制表示下的数位和（每一位上的数字相加），然后把数位和相等的数字放到同一个组中。

请你统计每个组中的数字数目，并返回数字数目并列最多的组有多少个。

 

示例 1：

输入：n = 13
输出：4
解释：总共有 9 个组，将 1 到 13 按数位求和后这些组分别是：
[1,10]，[2,11]，[3,12]，[4,13]，[5]，[6]，[7]，[8]，[9]。总共有 4 个组拥有的数字并列最多。

示例 2：

输入：n = 2
输出：2
解释：总共有 2 个大小为 1 的组 [1]，[2]。

示例 3：

输入：n = 15
输出：6

示例 4：

输入：n = 24
输出：5

 

提示：

• 1 <= n <= 10^4

思路：

模拟法，直接统计[1, n]里每个数的数位之和并放到字典 l 里，key 是数位之和，val是这个数位之和出现的次数。

然后判断 l 里出现次数并列最大的数位之和有几个。

时间复杂度：O（NlogN）

空间复杂度：O（logN）

class Solution(object):
    def countLargestGroup(self, n):
        """
        :type n: int
        :rtype: int
        """
        from collections import defaultdict
        l = defaultdict(int)   

        def helper(num):
            # 计算num数位之和，eg：输入34, 返回3 + 4 = 7
            s = 0
            while num:
                num, tmp = divmod(num, 10)
                s += tmp
            return s

        for num in range(1, n + 1):
            l[helper(num)] += 1
        
        mmax = max(l.values())
        return sum([1 for item in l.values() if item == mmax])