781. 森林中的兔子
题目:森林中有未知数量的兔子。提问其中若干只兔子 "还有多少只兔子与你(指被提问的兔子)颜色相同?" ,将答案收集到一个整数数组 answers 中,其中 answers 是第 i 只兔子的回答。给你数组 answers ,返回森林中兔子的最少数量
例子:
输入:answers =
输出: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(mapint)
var num int = 0
for index, value := range answers {
_ = index
fmt.Println(value)
v, isok := counter
_ = v
if value == 0 {
num = num + 1
} else if !isok || counter == 0 {
counter = value
num = num + value + 1
} else if counter != 0 {
counter = counter - 1
}
}
return num
}
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
页:
[1]