Question Answer

用的根号筛

#include<iostream>
#include<cmath>
using namespace std;
bool Is_prime(int x){
    if(x<2){
        return false;
    }
    int m=int(sqrt(x))+1;
    for(int i=2;i<m;i++){
        if(x%i==0){
            return false;
        }
    }
    return true;
}
int main(){
    int n,m,sum=0;
    cin>>n>>m;
    n++;
    int a[n],b[m];
    for(int i=1;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<m;i++){
        cin>>b[i];
        if(Is_prime(a[b[i]])){
            sum+=a[b[i]];
        }
    }
    cout<<sum;
    return 0;
}