写一个判别素数的函数在主函数输入一个整数

写一个判别素数的函数，在主函数输入一个整数，输出是否是素数的信息，本程序应当准备以下测试数据：17，34，2，1，0。

主函数就不写了

int isPrime(int n){
    if(n == 1) return 0;
    int i;
    for(i = 2;i*i <= n;i++){
        if(n % i == 0) return 0;
    }
    return 1;
}

#include <iostream>
using namespace std;

bool f (int n)
{
    for (int i = 2; i <= n / i; i ++)
        if (n % i == 0) return false;
    return true;
}
int main()
{
    int n;
    cin >> n;
    while (n --)
    {
        int x;
        cin >> x;
        if (f(x)) printf("%d is prime\n",x);
        else printf("%d is not prime\n",x);
    }
    return 0;
}

#include<stdio.h>
int isSuShu(int n) {
    if (n <2)
        return 0;
    int i;
    for (i = 2; i * i <= n; i++) {
        if (n % i == 0)
            return 0;
    }
    return 1;
}

int main()
{
    int n = 0;
    scanf("%d",&n);
    int is=isSuShu(n);
    if (is == 1)
    {
        printf("是素数！");
    }
    else {
        printf("不是素数！");
    }
    return 0;
}