【Leetcode 每日一题】781. 森林中的兔子

最新推荐文章于 2025-08-23 02:01:11 发布

冠位观测者

最新推荐文章于 2025-08-23 02:01:11 发布

阅读量1k

点赞数 15

CC 4.0 BY-SA版权

分类专栏： Leetcode Daily 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/2401_88859777/article/details/147358916

Leetcode Daily 专栏收录该内容

195 篇文章

订阅专栏

问题背景

森林中有未知数量的兔子。提问其中若干只兔子 “还有多少只兔子与你（指被提问的兔子）颜色相同?” ，将答案收集到一个整数数组 $an s w ers$ 中，其中 $an s w ers [i]$ 是第 $i$ 只兔子的回答。
给你数组 $an s w ers$ ，返回森林中兔子的最少数量。

数据约束

$\le answers.length \le 1000$
$\le answers[i] < 1000$

解题过程

找规律，考虑到要求数量最少，假设当前数组元素是 $i t e m$ ，它共有 $n u m$ 个等值元素，那么答案会增加 $⌈numitem+1⌉\lceil \frac{num}{item + 1} \rceil$ 。
建立数组元素与数量之间的哈希映射，再遍历计算结果即可。

具体实现

class Solution {
    public int numRabbits(int[] answers) {
        Map<Integer, Integer> count = new HashMap<>();
        for (int item : answers) {
            count.merge(item, 1, Integer::sum);
        }
        int res = 0;
        for (Map.Entry<Integer, Integer> entry : count.entrySet()) {
            int item = entry.getKey();
            int num = entry.getValue();
            res += (item + num) / (item + 1) * (item + 1);
        }
        return res;
    }
}