carloscn / structstudy

Leetcode daily trainning by using C/C++/RUST programming.
4 stars 1 forks source link

leetcode1742:盒子中小球的最大数量(maximum-number-of-balls-in-a-box) #273

Open carloscn opened 1 year ago

carloscn commented 1 year ago

问题描述

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

来源:力扣(LeetCode) 链接:https://leetcode.cn/problems/maximum-number-of-balls-in-a-box

carloscn commented 1 year ago

问题分析

use std::collections::HashMap;

fn pos_sum(val:i32) -> i32
{
    let mut vp:i32 = val;
    let mut ret:i32 = 0;

    while vp != 0 {
        ret += vp % 10;
        vp /= 10;
    }

    return ret;
}

pub fn count_balls(low_limit: i32, high_limit: i32) -> i32
{
    if high_limit < low_limit {
        return 0;
    }

    let mut ret:i32 = i32::MIN;
    let mut box_num:HashMap<i32, i32> = HashMap::new();

    for i in low_limit..high_limit + 1 {
        *box_num.entry(pos_sum(i)).or_default() += 1;
    }

    box_num.iter().fold(i32::MIN, |count, (bx, num)| {
        ret = count.max(*num); ret
    });

    return ret;
}
carloscn commented 1 year ago

code

https://review.gerrithub.io/c/carloscn/structstudy/+/556620 https://github.com/carloscn/structstudy/commit/1d7ed5caf130c5b38409cee5e579d661dd26e139