森林中,每個兔子都有顏色。其中一些兔子(可能是全部)告訴你還有多少其他的兔子和自己有相同的顏色。我們將這些回答放在 answers 數組里。
返回森林中兔子的最少數量。
示例:
輸入: answers = [1, 1, 2]
輸出: 5
解釋:
兩只回答了 “1” 的兔子可能有相同的顏色,設為紅色。
之后回答了 “2” 的兔子不會是紅色,否則他們的回答會相互矛盾。
設回答了 “2” 的兔子為藍色。
此外,森林中還應有另外 2 只藍色兔子的回答沒有包含在數組中。
因此森林中兔子的最少數量是 5: 3 只回答的和 2 只沒有回答的。
輸入: answers = [10, 10, 10]
輸出: 11
輸入: answers = []
輸出: 0
解題思路
將具有報出相同數量同類的兔子作為一組,遍歷一次數組,就可以知道每個組內缺了多少個兔子,再加上原數組中的兔子個數,就是森林中兔子的最少數量。
代碼
class Solution {public int numRabbits(int[] answers) {HashMap<Integer, Integer> map = new HashMap<>();for (int i = 0; i < answers.length; i++) {if(answers[i]==0) continue; //獨一無二的兔子map.put(answers[i],map.getOrDefault(answers[i],answers[i]+1)-1);if(map.get(answers[i])==0) map.remove(answers[i]);//這個組已經查完了,移除}int res=answers.length;for (Map.Entry<Integer, Integer> entry : map.entrySet()) {res+=entry.getValue();}return res;}
}