PTA试题—质数的后代（java）

题目：

在上一季里，曾提到过质数的孤独，其实从另一个角度看，无情隔膜它们的合数全是质数的后代，因为合数可以由质数相乘结合而得。
如果一个合数由两个质数相乘而得，那么我们就叫它是质数们的直接后代。现在，给你一系列自然数，判断它们是否是质数的直接后代。
输入格式:
第一行一个正整数T，表示需要判断的自然数数量
接下来T行，每行一个要判断的自然数
输出格式:
共T行，依次对于输入中给出的自然数，判断是否为质数的直接后代，是则输出Yes，否则输出No
输入样例:
在这里给出一组输入。例如：
4
3
4
6
12

输出样例:
在这里给出相应的输出。例如：
No
Yes
Yes
No

代码：

import java.util.Scanner;

public class Main {
    
    
    public static void main(String[] args) {
    
    
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();

        int[] arr=new int[n];
        for (int i = 0; i < n; i++) {
    
    
            arr[i]=input.nextInt();
        }
        int m;
        for (int i = 0; i < n; i++) {
    
    

            m=0;
            for (int j = 2;arr[i] != 1; j++) {
    
    
                while(arr[i]%j==0){
    
    
                    m++;
                    arr[i]=arr[i]/j;
                }
                if (m>2){
    
    
                    break;
                }
            }
            if (m==2){
    
    
                System.out.println("Yes");
            }
            else {
    
    
                System.out.println("No");
            }
        }

    }
}