zznuoj1555 : 双胞胎素数 C语言

#include<stdio.h>

#include<math.h>

int prime(int n);

int main(void)

{

 int T,M,N,i;

 scanf("%d",&T);

 for(i=1;i<=T;i++)

 {

  scanf("%d %d",&M,&N);

  if(prime(M)&&prime(N)&&abs(M-N)==2)

  {

   printf("YES\n");

  }

  else

  printf("NO\n");

 }

 return 0;

}

int prime(int n)

 {

  int k,i,flag=0;

  k=(int)sqrt(n);

  for(i=2;i<=k;i++)

  {

   if(n%i==0)

   {

    flag=1;                        

    break;

   }

  }

  if(flag==1||n==1)

  return 0;

  return 1;

 }