PAT乙级 1007 素数对猜想（C语言，更改判断素数的循环，减少运行时间）

我的思路

1求出n以内的所有素数，并存入数组；
2在数组中判断s[i+1]-s[i]是否等于2；

注意的地方：

刚开始在判断n是否为素数的算法中，我写的for循环是（2至n），运行结果显示超时；改为（2至sqrt（n））时才显示正确。

#include<stdio.h>
#include<math.h>
int main(){
 int n,k=0;
 int account=0; //素数对的计数器 
 scanf("%d",&n);
 int s[n];
 if(n<5)printf("0");
 else{
  for(int i=3;i<=n;i++){
   int flag=0; //标记是否为素数 
   for(int j=2;j<=sqrt(i);j++){ //判断是否为素数 
    if(i%j==0){
     flag=1;
     break;
    }
   }
   if(flag==0)s[k++]=i; // i为素数，存入数组s中 
   
  }
  
  for(int i=0;i<k-1;i++){ //遍历数组，判断相邻素数对是否为相差2 
   if(s[i+1]-s[i]==2)account++;
  }
  printf("%d",account);
 }
 
}