【PTA】【C语言】判断给定整数是否素数[加强版]

素数是指在大于1的自然数中,除了1和它本身以外,不能被其他自然数整除的数。本题要求编写程序，判断从键盘输入的数是否素数。如果是素数，输出XX is prime number.的信息，输出XX is not prime number.的信息。

编写程序要求：

编写一个函数IsPrime()来判断素数。如果是素数，函数的返回值就是该数，否则函数的返回值为0.

输入格式:

在一行中输入一个非负整数.

输出格式:

在一行输出XX is prime number. 或者 XX is not prime number.的信息。

输入样例1:

5

输出样例1:

5 is prime number.

输入样例2:

121

输出样例2:

121 is not prime number.

代码如下： 

#include <stdio.h>
#include <string.h>
#include <math.h>
int prime(int n)
{
    int i;
    if (n == 0 || n == 1)
        return 0;
    for (i = 2; i <= (int)sqrt(n); i++)
    {
        if (n % i == 0)
            return 0;
    }
    return n;
}
 
int main()
{
    int n;
    scanf("%d",&n);
    if (prime(n) == n)
        printf("%d is prime number.", n);
    else
        printf("%d is not prime number.", n);
    return 0;
}