1
class Solution {
2
  public int numRabbits(int[] answers) {
3
    HashMap<Integer, Integer> map = new HashMap<>();
4
    int count = 0;
5

6
    for (int ele : answers) {
7

8
      if (!map.containsKey(ele + 1)) {
9
        map.put(ele + 1, 1);
10
        count += ele + 1;
11
      } else if (map.get(ele + 1) == ele + 1) {
12
        map.put(ele + 1, 1);
13
        count += ele + 1;
14
      } else {
15
        int freq = map.get(ele + 1);
16
        map.put(ele + 1, freq + 1);
17
      }
18
    }
19

20
    return count;
21
  }
22
}