CSDN挑战赛 - 素因子集合

/**题目：小强最近在学初等数论，老师给他们出了一个课后习题，
 * 那就是给你两个正整数A，B（0<A,B<2^60）,判断他们的素因
 * 子集合是否相同，小强刚接触数论，想了好一会还是没能想出
 * 来，你能帮助他吗？ 输入描述： 输入包含多组测试数据，
 * 每组测试数据包含两个正整数A，B,以文件结束。 
 * 输出描述： 对于每组测试数据如果A和B的素因子集合相同则
 * 输出“YES”，否则输出“NO”。
 * 解决思路：
 * （1）先求因数；
 * （2）再判断是不是质数；
 * （3）比较A和B的质因数是不是完全相等
 * 
 */
import java.util.ArrayList;
import java.util.Scanner;

public class Test0413 {

	private static String judgeNum(int A, int B){
		ArrayList<Integer> aList = prime(A);
		ArrayList<Integer> bList = prime(B);
		
		if (aList.size() != bList.size()) {
			return "NO";
		}else {
			for (int i = 0; i <aList.size(); i++) {
				if (aList.get(i) != bList.get(i)) {
					return "NO";
				}
			}
			return "YES";
		}
	
	}
	
	private static ArrayList<Integer> prime(int a){
     ArrayList<Integer> aList = new ArrayList<Integer>();
		
		boolean isPrime = true;
		for (int i = 2; i <=a; i++) {
			if (a%i == 0) { //是A 的因数
				//判断是不是质数
				isPrime = true;
				for (int j = 2; j < i; j++) {
					if(i%j==0){
						isPrime = false;
					}
				}
				if (isPrime) {
					aList.add(i);
				}
			}
		}
		return aList;
	}
	
	
	public static void main(String args[]){
		
		Scanner reader = new Scanner(System.in);
		while(reader.hasNext()){
			int A = reader.nextInt();
			int B = reader.nextInt();
			System.out.println(judgeNum(A,B));
		}
		
	}
}