This code shows TLE error for question Prateek Loves Candy

#1

import java.util.*;
public class Main {
public static void main(String args[]) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int a[]=new int[n];
for(int i=0;i<n;i++){
a[i]=sc.nextInt();

	}
	for(int i=0;i<n;i++){
		if(a[i]==1){
			System.out.print("2");
			continue;
		}
		int c=a[i];
		int f=0;
		for(int  j=2;j<c*c;j++){
			boolean ch=true;
			for(int k=2;k<j;k++){
				if(j%k==0){
					ch=false;
					break;

				}
			}
			if(ch==true){
				f++;
			}
			if(f==c){
			System.out.println(j);
			break;
			}
		}
	}

}

}

#2

the constraint on number of test cases, It’s quite large because for every test case you are going to find the prime numbers upto that number and at last return the last prime number found, this will cause the TLE for big test cases.

  • So, to overcome this problem, let’s just store your test cases in an array.
  • Then find the maximum number from all of the testcases.
  • then, Use SOE(Sieve of Eratosthenes) to find the prime number upto that number and store that an the array.
  • After storing, simply loop over every test case that was stored previously in an array.
  • Print the value of n from the array storing primes.

That’s how we need not to find the prime number for every test case. :slight_smile:
try to understand this code here by debugging through it