Help me in solving AO21 problem

[vigneshwar2004]

My issue

expiain the logic

Learning course: Beginner DSA in Python
Problem Link: Candies Practice Problem in Beginner DSA in Python - CodeChef