781. 森林中的兔子▲▲

题目:
森林中,每个兔子都有颜色。其中一些兔子(可能是全部)告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 answers 数组里。

返回森林中兔子的最少数量。

示例:
输入: answers = [1, 1, 2]
输出: 5
解释:
两只回答了 “1” 的兔子可能有相同的颜色,设为红色。
之后回答了 “2” 的兔子不会是红色,否则他们的回答会相互矛盾。
设回答了 “2” 的兔子为蓝色。
此外,森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。

输入: answers = [10, 10, 10]
输出: 11

输入: answers = []
输出: 0

说明:
answers 的长度最大为1000。
answers[i] 是在 [0, 999] 范围内的整数。

分析:
报不同数字的兔子肯定具有不同的颜色,求兔子的最少总量就是要尽可能让更多的报相同数字的兔子配对。

Solution:
用unordered_map<int,int>mp记录下报相同数字的兔子的个数。
key为兔子报的数字,value为报该数字的兔子的个数。
那么:
kay==0时,报0的兔子肯定没有其他兔子与其有相同的颜色,兔子数量加mp[0]。
key!=0时,假如有一只兔子报了数字key,则最多有key-1个兔子可能与其具有相同的颜色。
*例如key=9,则value<=10时,它们都可能只有一种颜色,因为可能有0或几只兔子没有回答,当他们都是同一种颜色时,兔子总量最小。而当value>10时,他们肯定不止一种颜色。(如果都具有相同颜色,那么肯定报的就不是9了)。报key的兔子所占的颜色最少有(b-1)/(a+1)+1种。每种颜色有a+1个兔子。

class Solution {
public:
    int numRabbits(vector<int>& answers) {
        unordered_map<int,int>mp;
        for(auto a:answers)
            mp[a]++;
        int res=0;
        for(auto [a,b]:mp){
            if(a==0)
                res+=b;
            else
                res+=(a+1)*((b-1)/(a+1)+1);            
        }
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值