题目
你在一家生产小球的玩具厂工作,有 n 个小球,编号从 lowLimit 开始,到 highLimit 结束(包括 lowLimit 和 highLimit ,即 n == highLimit - lowLimit + 1)。另有无限数量的盒子,编号从 1 到 infinity 。
你的工作是将每个小球放入盒子中,其中盒子的编号应当等于小球编号上每位数字的和。例如,编号 321 的小球应当放入编号 3 + 2 + 1 = 6 的盒子,而编号 10 的小球应当放入编号 1 + 0 = 1 的盒子。
给你两个整数 lowLimit 和 highLimit ,返回放有最多小球的盒子中的小球数量。如果有多个盒子都满足放有最多小球,只需返回其中任一盒子的小球数量。
示例
示例 1:
输入:lowLimit = 1, highLimit = 10
输出:2
解释: 盒子编号:1 2 3 4 5 6 7 8 9 10 11
… 小球数量:2 1 1 1 1 1 1 1 1 0 0 … 编号 1 的盒子放有最多小球,小球数量为 2 。
示例 2:
输入:lowLimit = 5, highLimit = 15
输出:2
解释: 盒子编号:1 2 3 4 5 6 7 8 9 10 11
… 小球数量:1 1 1 1 2 2 1 1 1 0 0 … 编号 5 和 6 的盒子放有最多小球,每个盒子中的小球数量都是 2
。
示例 3:
输入:lowLimit = 19, highLimit = 28
输出:2
解释: 盒子编号:1 2 3 4 5 6 7 8 9 10 11
12 … 小球数量:0 1 1 1 1 1 1 1 1 2 0 0 … 编号 10 的盒子放有最多小球,小球数量为 2 。
提示:
1 <= lowLimit <= highLimit <= 105
思路
哈希表记录每个盒子中装的小球的数量即可(键为盒子编号,值为该编号中小球的数量),从lowLimit到highLimit遍历小球编号,求出对应盒子编号,加入哈希表中记录,最后返回哈希表中值的最大值即可
题解
class Solution: def countBalls(self, lowLimit: int, highLimit: int) -> int: # 哈希表 tmp = {} # 遍历小球编号并求对应盒子编号,记录在哈希表中 for i in range(lowLimit, highLimit+1): index = str(i) s = 0 for j in range(len(index)): s += int(index[j]) if tmp.get(s): tmp[s] += 1 else: tmp[s] = 1 # 返回哈希表值的最大值即可 return max(tmp.values())