1029 自定义函数处理素数

题目描述
写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息。
输入
一个数
输出
如果是素数输出prime 如果不是输出not prime
样例输入复制
97
样例输出复制
prime

---

运行超限

#include<iostream>
#include<cmath>
using namespace std;
bool isprime(int n)
{
    for (int i = 2; i <= sqrt(n); i++)
    {
        if (n % i == 0)
        {
            return false;
            return 0;
        }
    }
    return true;
}
int main()
{
    int n;
    cin >> n;
    if (isprime(n))
    {
        cout << "prime";
    }
    else
    {
        cout << "not prime";
    }
    return 0;
}

---

正确代码

#include <stdio.h>
 
void judgmentf(int a);
 
int main()
{
    int a;
    scanf("%d",&a);
 
    judgmentf(a);
 
    return 0;
}
 
void judgmentf(int a)
{
    int i;
 
    for(i=2;i<a;i++)
    {
        if(a%i==0)
        {
            printf("not prime");
            break;
        }
    }
 
    if(a==i)
    {
        printf("prime");
    }
}