135-candy.go

// Link: https://leetcode.com/problems/candy/

func max(a,b int) int {
    if a > b {
        return a
    }
    return b
}

func candy(ratings []int) int {
    n := len(ratings)
    ans := 0
    candies := make([]int, n)
    for i:=0; i<n; i++ {
        candies[i] = 1
    }
    for i:=1; i<n; i++ {
        if ratings[i] > ratings[i-1] {
            candies[i] = candies[i-1]+1
        }
    }
    ans += candies[n-1]
    for i:=n-2; i>=0; i-- {
        if ratings[i] > ratings[i+1] {
            candies[i] = max(candies[i], candies[i+1]+1)
        }
        ans += candies[i]
    }
    return ans
}