1029: [编程入门]自定义函数处理素数

解题思路:

定义一个求素数的函数并调用就好了~

注意事项:

是小于等于n不是小于n。

参考代码:

#include<bits/stdc++.h>
using namespace std;
int f(int n)
{
	for(int i=2;i*i<=n;i++)
	{
		if(n%i==0) return 0;
	}
	return 1;
}
int main()
{
	int n;
	cin>>n;
	if(f(n)==1) cout<<"prime";
	else if(f(n)==0) cout<<"not prime";
	return 0;
}