PAT——1099 性感素数

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

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

输入格式：

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

输出格式：

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

输入样例 1：

47

样例-1：">样例-1：">输出样例 1：

Yes
41

输入样例 2：

21

输出样例 2：

No
23

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

解题思路：按照要求判断一个数n是否是素数即可

如果n和n-6都是素数，则输出n-6；若n和n+6都是素数，则输出n+6；

若n不是素数，则要找n+1往后的最小性感素数，重复上面的步骤即可。

测试点2：在判断素数的时候，原本写的是if(n==0||n==1)  return 0出错了，改成if(n<2) return 0就过了，可能是因为后面会判断n-6可能会出现负数的情况。

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
int isPrime(int n)        //判断是否是素数
{ 
	
	if(n<2)                  //要判断n<2，后面n-6可能会出现负数的情况
	   return 0;
	for(int i=2;i<=sqrt(n);i++)
	{
		if(n%i==0)
		   return 0;
	}
	return 1;
}
int main()
{
	long long int n;
	cin>>n;
    if(isPrime(n)&&(isPrime(n-6)||isPrime(n+6)))
    {
    	if(isPrime(n-6))
        {
        	cout<<"Yes"<<endl;
        	cout<<n-6;
		}
		else
		{
			cout<<"Yes"<<endl;
			cout<<n-6;
		}
	}
    else              //n不是素数，要从n+1开始找最小性感素数
    {
    	cout<<"No"<<endl;
        long long int number;
    	for(number=n+1;;number++)
    	{
    		if(isPrime(number)==1)
    		{
    			if(isPrime(number+6)==1||isPrime(number-6)==1)
                {
    			   cout<<number;
                   break;
                }
			}
		}
	}
}