试题 算法提高 素数判断

题目描述：
编写一函数IsPrime，判断某个大于2的正整数是否为素数。
样例输入:
5
样例输出:
yes
样例输入:
9
样例输出:
no
注意：是素数输出yes，不是素数输出no，其中yes和no均为小写。

资源限制
时间限制：1.0s 内存限制：512.0MB

该题有判断素数的模板……
代码块：

import java.util.Scanner;
public class Main {
	public static void main(String[] args){
		Scanner sc = new Scanner(System.in);
		
		int n = sc.nextInt();
		if(su(n)){
			System.out.println("yes");
		}else{
			System.out.println("no");
		}
	}
	
	//判断是否是素数
	static boolean su(int n){
		
		for(int i = 2; i*i <= n; i++){
			if(n%i == 0){
				return false;
			}
		}
		
		return true;
	}
}