用函数判断是否是素数

#include<stdio.h>
int is_prime(int n);
int main()
{
    int n;
    printf("请输入一个整数：");
    scanf("%d",&n);
    if(is_prime(n)) printf(" %d 是一个素数\n",n);
    else printf(" %d 不是素数\n",n);
    return 0;
}
int is_prime(int n)
{
    int i;
    if(n==1) return 0;
    for(i=2;i*i<=n;i++)
	{
        if(n%i==0) return 0;
    }
    return 1;
}