自定义函数处理素数

import java.util.Scanner;

public class Main {
    // 判断一个数是否是素数的函数
    public static boolean isPrime(int n) {
        if (n <= 1) {
            return false;
        }
        if (n <= 3) {
            return true;
        }
        if (n % 2 == 0 || n % 3 == 0) {
            return false;
        }
        for (int i = 5; i * i <= n; i += 6) {
            if (n % i == 0 || n % (i + 2) == 0) {
                return false;
            }
        }
        return true;
    }

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

        if (isPrime(num)) {
            System.out.println("prime");
        } else {
            System.out.println("not prime");
        }
    }
}