Java计算素数

/*需求：素数
* 素数：就是除了可以被他自身和1整除外，没有其他整除的数
* 1.键盘输入要输出多少以内的素数
* 2.构造一个方法，将输入的数传进去
* 3.在方法中实现代码*/
public class PrimeNumber {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n;
        //定义一个集合，去存放素数
        ArrayList<Integer> arrayList = new ArrayList<>();
        System.out.print("请输入一个大于1正整数：");
        n = scanner.nextInt();
        //进行判断，如果n小于1，则不符合要求，结束程序
        if (n < 2){
            System.out.println("输入的数不符合要求");
            System.exit(0);
        }
        //从2开始一直到输入的数，判断有多少个素数
        for (int i = 2; i <= n; i++) {
            if (primeNumber(i)){
                arrayList.add(i);
            }
        }
            System.out.println(n+"以内的素数有：\n"+arrayList);
    }
//判断素数的方法
    private static boolean primeNumber(int i) {
        //如果是2和3直接返回
        if (i == 2 || i == 3)return true;
        else {
            //判断，如果这个数对它的平方根以下的数取余有任意一个数等于零，则不是素数
            for (int j = 2; j <= Math.sqrt(i); j++) {
                if (i % j == 0){
                    return false;
                }
            }
            return true;
        }
    }
}