PTA乙级_1007 素数对猜想 (20分)_C语言版

让我们定义d​n​​为：d​n​​=p​n+1​​−p​n​​，其中p​i​​是第i个素数。显然有d​1​​=1，且对于n>1有d​n​​是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的素数”。
现给定任意正整数N(<10​5​​)，请计算不超过N的满足猜想的素数对的个数。
输入格式:
输入在一行给出正整数N。
输出格式:
在一行中输出不超过N的满足猜想的素数对的个数。
输入样例:
20
输出样例:
4

思路：首先定义函数prime判断是否为素数，然后核心代码就是判断prime(i)&&prime(i+2)

#include<stdio.h>
#include<math.h>
int prime(int n){
	int i;
	for (i=2;i<=sqrt(n);i++){
		if (n%i==0){
			return 0;
		}
	}
	if (i>sqrt(n)){
		return 1;
	}
}
int main(){
	int i,N;
	int count = 0;
	scanf("%d",&N);
	for (i=2;i<N-1;i++){
		if (prime(i)&&prime(i+2)){
			count++;
		}
	}
	printf("%d",count);
	return 0;
}