781. 森林中的兔子 Java-白红宇

781. 森林中的兔子 Java

阅读量：351 次

发布时间：2019-03-04

本文共 1457 字，大约阅读时间需要 4 分钟。

为了解决这个问题，我们需要根据兔子的回答来计算森林中兔子的最少数量。每个兔子都会给出它们看到的同颜色兔子的数量，我们需要利用这些信息来推断出森林中兔子的总数。

方法思路

输入处理：首先检查输入数组是否为空，如果为空，直接返回0。

排序数组：对输入数组进行排序，以便更好地处理相同数量的回答。

统计回答次数：遍历排序后的数组，统计每个数量出现的次数。

计算兔子数量：对于每个数量，根据其出现次数和数量的关系，计算对应颜色的兔子数量。

如果出现次数等于数量+1，则该颜色的兔子数量就是数量+1。

如果出现次数小于数量+1，则该颜色的兔子数量是数量+1。

如果出现次数大于数量+1，则该颜色的兔子数量是出现次数。

求和：将所有颜色的兔子数量相加，得到森林中兔子的最少数量。

解决代码

import java.util.*;
public class Solution {
    public static int numRabbits(int[] answers) {
        if (answers.length == 0) return 0;
        Arrays.sort(answers);
        int count = 0;
        Map
   
     map = new HashMap<>();
        int loc = 0;
        for (int i = 1; i < answers.length; i++) {
            if (answers[i] != answers[i - 1]) {
                map.put(answers[i - 1], i - loc);
                loc = i;
            }
        }
        map.put(answers[answers.length - 1], answers.length - loc);
        
        for (Entry
    
      entry : map.entrySet()) {
            int key = entry.getKey();
            int value = entry.getValue();
            if (value == key + 1) {
                count += value;
            } else if (value < key + 1) {
                count += key + 1;
            } else {
                count += key + 1 + (value - (key + 1));
            }
        }
        return count;
    }
}