盒子中小球的最大数量
难度:简单
你在一家生产小球的玩具厂工作,有 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 。
解法一、哈希表
思路:
遍历所有的小球,对于编号为 i 的小球,计算它应该放入的盒子编号 box,使用哈希表 box 记录每个盒子中的小球数量,返回遍历结束后 box 中小球数量最大的盒子对应的小球数量即可。
时间复杂度:
O
(
n
l
o
g
h
i
g
h
L
i
m
i
t
)
O(nloghighLimit)
O(nloghighLimit),其中
n
=
h
i
g
h
L
i
m
i
t
−
l
o
w
L
i
m
i
t
+
1
n=highLimit−lowLimit+1
n=highLimit−lowLimit+1。
空间复杂度:
O
(
l
o
g
h
i
g
h
L
i
m
i
t
)
O(loghighLimit)
O(loghighLimit)。假设
h
i
g
h
L
i
m
i
t
highLimit
highLimit 的十进制位数为
x
x
x,那么可能使用的盒子编号数目不超过
10
×
x
10×x
10×x,因此空间复杂度为
O
(
l
o
g
h
i
g
h
L
i
m
i
t
)
O(loghighLimit)
O(loghighLimit)。
class Solution:
def countBalls(self, lowLimit: int, highLimit: int) -> int:
maxs = 0
box = dict()
for i in range(lowLimit, highLimit+1):
sums = sum(map(int, str(i)))
if box.get(sums):
box[sums] = box.get(sums) + 1
else:
box[sums] = 1
if maxs < box[sums]:
maxs = box[sums]
return maxs
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/maximum-number-of-balls-in-a-box/solutions