C++ :

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
/*
写一个判断素数的函数，
在主函数输入一个整数，输出是否是素数的消息。
*/

int f1(int a)
{
    int i,flag=1;
    if(a==2||a==3)
    printf("prime");
    else
    {
        for(i=3;i<=sqrt(a);i++)
        {
            if(a%i==0)
            {
                flag=0;
                break;
            }
        }
        if(flag==1)
        printf("prime");
        else
        printf("not prime");
    }
    return 0;
}


int main(int argc, char *argv[])
{
  int a;
  scanf("%d",&a);
  f1(a);
  

  return 0;
}