找素数并保存到数组中

#include<math.h>
int  fun(int n,int m,int a[])
{
    int i,k=0;
    for(i=n;i<=m;i++)
    {
        if(prime(i)){
            a[k]=i;
            k++;
        }
    }
    return k;
}

int prime(int n)
{
    int j;
    if(n<2) return 0;
    else if(n==2) return 1;
    else
        for(j=2;j<=sqrt(n)+1;j++)
            if(n%j==0) return 0;
    return 1;    
}