题目:森林中有未知数量的兔子。提问其中若干只兔子 "还有多少只兔子与你(指被提问的兔子)颜色相同?" ,将答案收集到一个整数数组 answers 中,其中 answers 是第 i 只兔子的回答。
给你数组 answers ,返回森林中兔子的最少数量
例子:
输入:answers = [1,1,2]
输出:5
解释:
两只回答了 "1" 的兔子可能有相同的颜色,设为红色。
之后回答了 "2" 的兔子不会是红色,否则他们的回答会相互矛盾。
设回答了 "2" 的兔子为蓝色。
此外,森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5 只:3 只回答的和 2 只没有回答的。
思路:
遍历answers
如果 i == 0:则此颜色的有且只有他一个:num+1
如果 i !=0:此数字是第一次出现,或者这个数字的值大于前面出现的总次数,说明是另一种颜色的相同数字:num+i+1
如果 i 不是第一次出现:则i的计数器要减1,减到0重复第二步。
go代码:- func numRabbits(answers []int) int {
- counter := make(map[int]int)
- var num int = 0
- for index, value := range answers {
- _ = index
- fmt.Println(value)
- v, isok := counter[value]
- _ = v
- if value == 0 {
- num = num + 1
- } else if !isok || counter[value] == 0 {
- counter[value] = value
- num = num + value + 1
- } else if counter[value] != 0 {
- counter[value] = counter[value] - 1
- }
- }
- return num
- }
复制代码
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作! |