用c语言或者c++编程。

写一个判素数的函数，在主函数输入一个整数，输出是否为素数的信息。

#include <stdio.h>
int isPrime(int num){
    if(num == 1) return 0;
    int i;
    for (i = 2; i < num; i++)
    {
        if (num % i == 0)
        {
            return 0;
        }
    }
    return 1;
}
int main()
{
    int num = 0;
    scanf("%d", &num);
    if(isPrime(num)){
        printf("Yes");
    }else{
        printf("No");
    }
    return 0;
}