蒟蒻刚学OI#3WA,求助QwQ
查看原帖
蒟蒻刚学OI#3WA,求助QwQ
632635
WOERDESUGX楼主2022/10/13 08:16
// 
#include<bits/stdc++.h>
using namespace std;
int n,k,prime[1005],vis[1005],f=0;
int kk=1,ans=0;
void Prime(int n)
{
    for(int i=2;i<=n;i++) {
        if(!vis[i]) {
            prime[++prime[0]]=i;
            kk++;
        }
        for(int j=1;i*prime[j]<=n+1;++j) {
            vis[i*prime[j]]=1;
            if(i%prime[j]==0) break;
        }
    }
}
int main()
{
    cin>>n>>k;
    Prime(n);
    for(int i=1;i<kk;++i) {
        for(int j=1;j<i-1;++j) {
            if(prime[j]+prime[j+1]+1==prime[i]) {
                ans++;
            }
            if(ans==k) {
                cout<<"YES";
                return 0;
            }
        }
    }
    cout<<"NO";
    return 0;
}
2022/10/13 08:16
加载中...