管管的幸运数字（线性筛）

思路：这个题暴力也可以过，而我是先筛选出素数来，然后对于每次询问再去判定n是不是素数，如果不是再去找比它小和比它大的素数。最终找到最近的那个。

#include <bits/stdc++.h>
using namespace std;
int prime[100010];
void is_prime(int x){
    
    
    for(int i=2;i<=sqrt(x);i++){
    
    
        if(prime[i]==0){
    
    
            for(int j=i*i;j<=x;j+=i){
    
    
                prime[j] = 1;
            }
        }
    }
}

int main()
{
    
    
    int t; cin>>t;
    is_prime(100000);
    while(t--){
    
    
        int n; cin>>n;
        if(prime[n]==0) cout<<"YES"<<endl;
        else {
    
    
            int cnt1=0,cnt2=0;
            for(int i=n;;i++){
    
    
                if(prime[i]==0){
    
    
                    cnt1 = i;
                    break;
                }
            }
            for(int i=n;;i--){
    
    
                if(prime[i]==0){
    
    
                    cnt2 = i;
                    break;
                }
            }
            int ans = min(abs(n-cnt1),abs(n-cnt2));
            cout<<ans<<endl;
        }
    }
    return 0;
}

开始的时候数组开的是10005wa了，后面加大一些就过了。下次一定注意范围。