Prime visit array question number 1

#1


only 2 test cases are running out of 3. Cn you plz tell me how to optimize it