YTU OJ 1049: 判断素数

Description

写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息。

Input

一个数

Output

如果是素数输出prime 如果不是输出not prime

Sample Input

97

Sample Output

prime

HINT

  主函数已给定如下，提交时不需要包含下述主函数

/*  C代码  */

int main(){

    int flag,n;

    int is_prime(int);   

    scanf("%d",&n);

    flag=is_prime(n);

    if(flag==1)

        printf("prime\n");

    else

        printf("not prime\n");

    return 0;

}

/*  C++代码  */

int main(){

    int flag,n;

    int is_prime(int);

    cin>>n;

    flag=is_prime(n);

    if(flag==1)

        cout<<"prime"<<endl;

    else

        cout<<"not prime"<<endl;

    return 0;

}

Source

freeproblemset edit by lyh

【注意】：提交时头文件也要提交

【AC代码】：

#include <iostream>
using namespace std;
int is_prime(int n)
{
    for(int i=2;i<n;i++)
        if(n%i==0)
        return 0;
    return 1;
}