c加加编写程序希望懂的人解答

.写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息。

#include<iostream>
using namespace std;
bool isPrime(int n) {
    for(int i = 2; i * i <= n; i++) {
        if(n % i == 0)
            return false;
    }
    return true;
}
int main() {
    int n;
    cin>>n;
    if(isPrime(n)){
        cout<<"Yes";
    }else{
        cout<<"No"; 
    }
    return 0;
}