*Leetcode 781. Rabbits in Forest | 思维题

最新推荐文章于 2021-01-22 16:34:21 发布

Z-Pilgrim

最新推荐文章于 2021-01-22 16:34:21 发布

阅读量436

点赞数

分类专栏： LeetCode题解思维题

本文链接：https://blog.csdn.net/u011026968/article/details/79418880

版权

LeetCode题解同时被 2 个专栏收录

271 篇文章 0 订阅

订阅专栏

思维题

30 篇文章 0 订阅

订阅专栏

https://leetcode.com/contest/weekly-contest-71/problems/rabbits-in-forest/

挺有意思的题。找几个例子试一试就能过掉

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