28 lines
518 B
Go
28 lines
518 B
Go
|
package main
|
||
|
|
|
||
|
|
func candy(ratings []int) int {
|
||
|
|
count := 1
|
||
|
|
for idx := 0; idx < len(ratings)-1; {
|
||
|
|
for idx < len(ratings)-1 && ratings[idx+1] == ratings[idx] {
|
||
|
|
count++
|
||
|
|
idx++
|
||
|
|
}
|
||
|
|
next_inc := 1
|
||
|
|
for idx < len(ratings)-1 && ratings[idx+1] > ratings[idx] {
|
||
|
|
next_inc++
|
||
|
|
idx++
|
||
|
|
count += next_inc
|
||
|
|
}
|
||
|
|
next_dec := 1
|
||
|
|
for idx < len(ratings)-1 && ratings[idx+1] < ratings[idx] {
|
||
|
|
count += next_dec
|
||
|
|
next_dec++
|
||
|
|
idx++
|
||
|
|
}
|
||
|
|
if next_dec > next_inc {
|
||
|
|
count += next_dec - next_inc
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return count
|
||
|
|
}
|