![]()
1 def candy(ratings):
2 result=[1 for i in range(len(ratings))]
3 for i in range(len(ratings)-1):
4 if ratings[i]>ratings[i+1] and result[i]<=result[i+1]:
5 result[i]+=result[i+1]-result[i]+1
6 if ratings[i]<ratings[i+1] and result[i]>=result[i+1]:
7 result[i+1]+=result[i]-result[i+1]+1
8 for j in range(len(ratings)-1,0,-1):
9 if ratings[j]>ratings[j-1] and result[j]<=result[j-1]:
10 result[j]+=result[j-1]-result[j]+1
11 if ratings[j]<ratings[j-1] and result[j]>=result[j-1]:
12 result[j-1] += result[j ] - result[j-1] + 1
13 print(result)
14 return sum(result)
