LeetCode 1742. 盒子中小球的最大数量 Java实现&Python实现

LeetCode 1742. 盒子中小球的最大数量 Java&Python实现

你在一家生产小球的玩具厂工作，有 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 。

链接：https://leetcode-cn.com/problems/maximum-number-of-balls-in-a-box.

Java：

	public int countBalls(int lowLimit, int highLimit) {

		// 使用HashMap存储盒子对应的小球数量，key为盒子编号，value为小球数量
        Map map = new HashMap<Integer, Integer>();

        for(int i = lowLimit; i<= highLimit; i++) {
            int j = i;
            int tmp = 0;
            // 计算当前小球 i 应该放入的盒子编号tmp
            while (j >= 10) {
                tmp += j % 10;
                j /= 10;
            }
            tmp+=j;
            if(!map.containsKey(tmp))	// 盒子为空
                map.put(tmp, 1);
            else
                map.replace(tmp, (int)map.get(tmp)+1);	// 盒子已经有球了
        }
        
        // 这里寻找 max(map.values()),手写了一个for循环，有点拉跨，还有优化的空间
        int tmp = 0;
        for (Object i: map.values()) {
            if (tmp<(int)i)
                tmp = (int) i;
        }
        return tmp;
    }

Python：

# 整体思路和上面的Java差不多，只是这里使用的dict字典，return时可以调用python内置的max函数得到dic.values()的最大值
    def countBalls(self, lowLimit: int, highLimit: int) -> int:
        dic = {}
        for i in range(lowLimit, highLimit + 1):
            tmp = 0
            while i >= 10:
                tmp += i % 10
                i = i // 10
            tmp += i
            if tmp not in dic.keys():
                dic.setdefault(tmp, 1)
            else:
                dic.update({tmp: dic.get(tmp) + 1})

        return max(dic.values())  # 返回盒子中小球数量