被任何数整除用JAVA怎么表达

输入一个数，判断是否为质数，输入质数的那个数

import java.util.*;
public class main {
public static void main(String args[])
{
System.out.println("请输入：");
Scanner reader=new Scanner(System.in);
int num= reader.nextInt();
//num<=3时，质数有2和3
if (num <= 3&&num>1) {
System.out.println(num);
}else{
boolean flag =true;
//当n>3时，质数无法被比它小的数整除
for(int i = 2; i < n; i++){
if (n % i == 0) {
flag = false;
}
}
if(flag){
System.out.println(num);
}
}

}

import java.util.Scanner;

public class Test {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int num = scanner.nextInt();
        boolean flag = true;
        for (int i = 2; i <= Math.sqrt(num); i++) {
            if (num % i == 0) {
                flag = false;
                break;
            }
        }
        System.out.println(num + (flag ? "是" : "不是") + "质数");
    }
}

测试示例：

设置循环，循环变量从2到该数的平方根
循环里判断该数除以循环变量的余数是否为0