I'm getting tle in test cases

this will also give you TLE as you are not simplifying the question but make it more complex

first observe what the giving code is doing and then comeup with best efficient way to do the same

you can use this code for reference purpose
Reference Code

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.