PTA 乙级 1099 性感素数

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

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

输入格式：

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

输出格式：

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

输入样例 1：

47

输出样例 1：

Yes
41

输入样例 2：

21

输出样例 2：

No
23

#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
ll zhishu(ll n)
{
    int i=2;
    if(n<=1)
        return 0;
    for(;i<=sqrt(n);i++)
        if(n%i==0)
            return 0;
    return 1;
}
void sexs(ll m)
{
    if(zhishu(m)){
        if(zhishu(m-6)){
            cout<<"Yes"<<endl;
            cout<<m-6;
        }
        else if(zhishu(m+6))
        {
            cout<<"Yes"<<endl;
            cout<<m;
        }
        else sexs(m+1);
    }
    else
    {
        int flag=1;
        cout<<"No"<<endl;
        int s[1];
        int z=m;
        while(flag){
            for(int k=0;k<1;z++)
                if(zhishu(z))
                    s[k++]=z;
            if(zhishu(s[0]-6)||zhishu(s[0]+6)){
                cout<<s[0];
                flag=0;
            }
        }
    }
}
int main()
{
    long long int n;
    cin>>n;
    sexs(n);
}