题目大意:给定一个数组,代表每个小孩的评分,相邻的小孩如果评分高的得到更多的糖果,每个小孩初始时有一个糖果。
这是一道贪心题。刚开始以为遍历一遍就如果比前一个高的话就加1,否则减1。显然当数组减小的时候是无法保证其数量是正确的。什么意思呢?看下面我画的丑图:
按照我的想法如果遍历到第五个数字那么应该有4-1=3个糖果,但实际只能有一个。
正确做法:左遍历+右遍历
左遍历:如果后一个比前一个大的话那么等于前一个+1,否则等于1.
右遍历:思路一样,不过需要取每个位置的最大值,再来看一下我的丑图:
搞定:
func candy(ratings []int) int {
n := len(ratings)
left := make([]int,n)
for i:=0; i<n;i++{
if i >0 && ratings[i] > ratings[i-1]{
left[i] = left[i-1]+1
}else{
left[i] = 1
}
}
right :=0
res :=0
for i:= n-1;i>=0;i--{
if i < n-1 && ratings[i] > ratings[i+1]{
right++
}else{
right = 1
}
res += max(left[i], right)
}
return res
}
func max(a int, b int)int {
if a> b{
return a
}
return b
}
附上一张评论区看到的评论,笑拉了哈哈哈