g0701_0800.s0781_rabbits_in_forest.Solution Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of leetcode-in-java Show documentation
Show all versions of leetcode-in-java Show documentation
Java-based LeetCode algorithm problem solutions, regularly updated
package g0701_0800.s0781_rabbits_in_forest;
// #Medium #Array #Hash_Table #Math #Greedy
public class Solution {
public int numRabbits(int[] answers) {
int[] counts = new int[1001];
for (int element : answers) {
counts[element]++;
}
int answer = counts[0];
for (int i = 1; i <= 1000; i++) {
if (counts[i] > 0) {
int rabbitsInPartialGroup = counts[i] % (i + 1);
int rabbitsInCompleteGroups = counts[i] - rabbitsInPartialGroup;
answer += rabbitsInCompleteGroups;
if (rabbitsInPartialGroup > 0) {
answer += (i + 1);
}
}
}
return answer;
}
}