pat basic 1099 性感素数

“性感素数”是指形如 (p, p+6) 这样的一对素数。之所以叫这个名字，是因为拉丁语管“六”叫“sex”(即英语的“性感”)。（原文摘自 http://mathworld.wolfram.com/SexyPrimes.html）

现给定一个整数，请你判断其是否为一个性感素数。

输入格式：

输入在一行中给出一个正整数 N (≤10^8)。

输出格式：

若 N 是一个性感素数，则在一行中输出 Yes，并在第二行输出与 N 配对的另一个性感素数（若这样的数不唯一，输出较小的那个）。若 N 不是性感素数，则在一行中输出 No，然后在第二行输出大于 N 的最小性感素数。

输入样例 1：

47

输出样例 1：

Yes
41

输入样例 2：

21

输出样例 2：

No
23

解题思路：

对这道题来说可能构造一些函数会来的方便一些，另外在解题的时候要注意一些细节。

在yes的时候，要输出N的伙伴，no的时候，要输出“另一个N"。

#include <stdio.h>
#include <stdbool.h>

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

int check(int n) {
	int p1, p2;
	p1 = n - 6; p2 = n + 6;
	if ( isPrime(p1) )
		return p1;
	else if ( isPrime(p2) )
		return p2;
	else
		return 0;
}

int main(int argc, const char *argv[]) {
	int N, i, p;
	if ( scanf("%d", &N)==EOF ) printf("error\n");

	for ( i=N; ; ++i ) {
		if ( isPrime(i) ) {
			if ( p = check(i) ) {
				i == N ? printf("Yes\n%d", p) : printf("No\n%d", i);
				break;
			}
		}
	}

	return 0;
}