781. 森林中的兔子

题目链接：https://leetcode-cn.com/problems/rabbits-in-forest/

class Solution {
public:
    int numRabbits(vector<int>& answers) {
        int cnt[1000];
        memset(cnt, 0, 4000);
        int ans = 0;
        for(int i : answers){
            if(0 == cnt[i]){
                cnt[i] = i;
                ans += i + 1;
            }
            else cnt[i]--;
        }
        return ans;
    }
};