class Solution: def candy(self, ratings: List[int]) -> int: res = 1 pre = 1 fir = 1 j = 0 for i in range(1, len(ratings)): if ratings[i] > ratings[i - 1]: pre += 1 res += pre j = i fir = pre elif ratings[i] == ratings[i - 1]: res += 1 j = i pre = 1 fir = pre else: if pre == 1: res += i - j + int(i - j >= fir and ratings[j] > ratings[j+1]) else: res += 1 pre = 1 return res