Candy LeetCode solution in Python

1
# Runtime: 175 ms (Top 92.56%) | Memory: 16.8 MB (Top 75.67%)
2
class Solution:
3
    def candy(self, ratings: List[int]) -> int:
4
        n = len(ratings)
5
        temp = [1] * n
6

7
        for i in range(1, n):
8
            if ratings[i] > ratings[i - 1]:
9
                temp[i] = temp[i - 1] + 1
10
        if n > 1:
11
            if ratings[0] > ratings[1]:
12
                temp[0] = 2
13

14
        for i in range(n - 2, -1, -1):
15
            if ratings[i] > ratings[i + 1] and temp[i] <= temp[i + 1]:
16
                temp[i] = temp[i + 1] + 1
17

18
        return sum(temp)