Determining T_prime nos

2nd ans 3rd test cases shows TLE

A number is a T prime if it is a perfect square
and its square root is a prime number.
Please use the above written logic to solve the problem.

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.