蓝桥杯算法提高VIP-质数的后代 （Java代码）

解题思路:
不知道哪里错了，有大佬帮忙看一下吗



注意事项:





参考代码:

import java.util.ArrayList;
import java.util.Scanner;
public class 质数的后代 {
 public static void main(String[] args) {
  Scanner in=new Scanner(System.in);
  int n=in.nextInt();
  ArrayList<Integer> kk= new ArrayList<Integer>();
   for(int i=2;i<=105;i++){
   int j=2;
   while(j<i){
    if(i%j==0)break;
    j++;
   }
   if(i==j)kk.add(i);
  }
   int a[]=new int[n];
   for(int i=n-1;i>=0;i--){
   a[i]=in.nextInt();
  }
  for(int i=n-1;i>=0;i--){
   int j=2;int sum=0;boolean result=true;
   while(j<=a[i]){
    while(a[i]%j==0&&result==true){
     a[i]/=j;sum++;
     //判断i是不是质数
     if(!kk.contains(i)||sum>2)
     {
     result=false;
     }
    }
    j++;
   }
   if(a[i]==1&&sum==2){
    System.out.println("YES");
   }
   else {
    System.out.println("NO");
   }
  
  }
 }
}