Is my logic correct as it is showing tle for 2-3 cases

#1

i tried whole coding implementation by myself after just taking an idea of logic

#2

@akshitmehta
Your code seems fine with an 0(n) approach
Might be an infinite loop somewhere
Just for reference

#3

I hope I’ve cleared your doubt. I ask you to please rate your experience here
Your feedback is very important. It helps us improve our platform and hence provide you
the learning experience you deserve.

On the off chance, you still have some questions or not find the answers satisfactory, you may reopen
the doubt.