Leetcode 781. 森林中的兔子 (华为笔试题/数学题目)

 

这里要做的就是尽量将兔子分到同一组里，而尽量少的统计兔子个数。

 

兔子分到同一组的必要条件是，回答的个数相同。

因此首先，遍历一遍数组统计个数和频率，分别用ai和bi表示

该组中对应的兔子数量，至少是 bi/(ai+1) * (ai+1)上取整，即有多少组兔子，然后每一组都有ai+1个。

然后用一个技巧将上取整变为下取整，就能一行代码实现。

class Solution {
public:
    int numRabbits(vector<int>& answers) {
        unordered_map<int, int> cnt;
        int res = 0;
        for(auto x : answers){
            cnt[x]++;
        }
        for(auto [x, y] : cnt){
            res += (y + x) / (x + 1) * (x + 1);
        }
        return res;
    }
};