leetcode 781. 森林中的兔子

781. 森林中的兔子

前几天HW笔试原题……

class Solution {
public:
    int numRabbits(vector<int>& answers) {
        vector<bool> visited(answers.size(), false);
        int ans = 0;
        for(int ii = 0; ii < answers.size(); ii++) {
            if(visited[ii]) continue;
            for(int jj = ii+1, cnt = 0; jj < answers.size() && cnt < answers[ii]; jj++) {
                if(answers[jj] == answers[ii]) {
                    cnt++;
                    visited[jj] = true;
                }
            }
            ans += answers[ii] + 1;
        }
        return ans;
    }
};