TLE showing. Check for improvement

@igarg145 do right shift outside if loop
and dont forget to mark the doubt is cleared :smiley: