编写一个完整无错的源程序不用c++

编写程序，检查用户从键盘输入的整数是否为素数，并输出对应提示语句。

#include <stdio.h>
int isprime(int n)
{
    int i;
    if(n<2)
        return 0;
    for(i=2;i*i<=n;++i)
        if(n%i==0)
            return 0;
    return 1;
}
int main(void)
{
    int n;
    scanf("%d",&n);
    if(isprime(n))
        puts("YES");
    else
        puts("NO");
    return 0;
}