盒子中小球的最大数量

113 阅读2分钟

盒子中小球的最大数量

题目

原题链接:盒子中小球的最大数量

你在一家生产小球的玩具厂工作,有 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 <= 10^5

题解

用map存储<盒子编号,盒子中球的数量>,然后遍历lowLimit至highLimit之间的数量,取出各个位的和当作key,然后让该key的value加1,最后取出value的最大值即可。

class Solution {
    public int countBalls(int lowLimit, int highLimit) {
        //<盒子编号,盒子中球的数量>
        Map<Integer, Integer> map = new HashMap<>();
        //求最大值
        int max = 0;
        for (int i = lowLimit; i <= highLimit; i++) {
            //计算出key值
            int key = getSum(i + "");
            int value = map.getOrDefault(key, 0) + 1;
            //取最大值
            if (value > max) {
                max = value;
            }
            map.put(key, value);
        }
        return max;
    }

    //求出每位之和
    public int getSum(String str) {
        int sum = 0;
        for (int j = 0; j < str.length(); j++) {
            sum += str.charAt(j) - '0';
        }
        return sum;
    }
}