青训营 X 豆包MarsCode 技术训练营--小C点菜问题

题目

小C来到了一家餐馆，准备点一些菜。

已知该餐馆有 nn 道菜，第 ii 道菜的售价为 wiwi​。

小C准备点一些价格相同的菜，但小C不会点单价超过 mm 的菜。

小C想知道，自己最多可以点多少道菜？
测试样例

样例1：

输入：m = 6, w = [2, 3, 3, 6, 6, 6, 9, 9, 23]
输出：3

样例2：

输入：m = 4, w = [1, 2, 4, 4, 4]
输出：3

样例3：

输入：m = 5, w = [5, 5, 5, 5, 6, 7, 8]
输出：4

思路

我们可以使用一个数组来存储每道菜的价格。为了方便统计每种价格的菜的数量，我们可以使用一个哈希表（在Java中可以使用 HashMap）来记录每种价格的菜的数量。
过滤价格：首先，遍历价格数组，只保留价格不超过 m 的菜。
统计数量：使用哈希表统计每种价格的菜的数量。
找出最大值：遍历哈希表，找出数量最多的那种价格的菜的数量。

代码

import java.util.HashMap;
public class Main {
public static long solution(int m, int[] w) {
// 创建一个哈希表来统计每种价格的菜的数量
HashMap<Integer, Integer> priceCount = new HashMap<>();

    // 遍历价格数组，统计每种价格的菜的数量
    for (int price : w) {
        if (price <= m) {
            // 统计每种价格的菜的数量
            priceCount.put(price, priceCount.getOrDefault(price, 0) + 1);
        }
    }
    
    // 找出数量最多的那种价格的菜的数量
    int maxCount = 0;
    for (int count : priceCount.values()) {
        // 更新最大数量
        if (count > maxCount) {
            maxCount = count;
        }
    }
    
    return maxCount;
}

public static void main(String[] args) {
    System.out.println(solution(6, new int[]{2, 3, 3, 6, 6, 6, 9, 9, 23}) == 3);
    System.out.println(solution(4, new int[]{1, 2, 4, 4, 4}) == 3);
    System.out.println(solution(5, new int[]{5, 5, 5, 5, 6, 7, 8}) == 4);
}

}

priceCount.put(price, priceCount.getOrDefault(price, 0) + 1);
假设 priceCount 初始为空，当前遍历到的价格 price 为 3：
第一次遇到价格 3：
priceCount.getOrDefault(3, 0) 返回 0（因为 3 不存在）。
0 + 1 结果为 1。
priceCount.put(3, 1) 将 (3, 1) 存储到 priceCount 中。
第二次遇到价格 3：
priceCount.getOrDefault(3, 0) 返回 1（因为 3 已经存在）。
1 + 1 结果为 2。
priceCount.put(3, 2) 将 (3, 2) 存储到 priceCount 中。