No.95 - LeetCode781 - 一些兔子说同色有几个求最小兔子数

最新推荐文章于 2024-06-17 19:54:44 发布

原创最新推荐文章于 2024-06-17 19:54:44 发布 · 197 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用贪心算法解决兔子颜色问题的方法。通过统计每种答案出现的次数，利用贪心策略计算最少需要多少只不同颜色的兔子才能满足所有兔子的回答。这种方法不仅高效，而且能够确保解决方案的正确性。

贪心，回答不同的兔子颜色一定不同。
回答相同的兔子，颜色不一定相同，划分即可。

class Solution {
public:
    int numRabbits(vector<int>& answers) {
        map<int,int> mp;
        int N = answers.size();
        for(int i=0;i<N;i++) mp[answers[i]]++;
        int ans = 0;
        for(map<int,int>::iterator p=mp.begin();p!=mp.end();p++){
            ans += (p->second + p->first)/ (p->first+1)*(p->first+1);
        }
        return ans;
    }
};