请设计一个函数fun（），它的功能是判断pm是否是素数。若pm是素数，返回1；若不是素数，返回0。pm的值由主函数从键盘读入

请设计一个函数fun（），它的功能是判断pm是否是素数。若pm是素数，返回1；若不是素数，返回0。pm的值由主函数从键盘读入

#include <stdio.h>
int  fun(int pm)
{
    if(pm<2)
        return 0;
    for(int i=2;i*i<=pm;i++)
        if(pm%i==0)
            return 0;
    return 1;
}
int main()
{
    int pm;
    scanf("%d",&pm);
    int r = fun(pm);
    if(r==1)
          printf("%d是素数",pm);
    else
          printf("%d不是素数",pm);
    return 0;
}

int isprime(int m)
{
    int i=0;
    int k=(int)sqrt( (double)m );
    for(i=2;i<=k;i++)
        if(m%i==0) break;
        if(i>k)
            return 1;
        else
            return 0;
}

int fun(int x) //判断x是否为素数
{ int i;
for(i=2;i<x;i++)
if(x%i==0) break;
if(i==x) return 1;
else return 0;
}