题解

判断，循环

#include<bits/stdc++.h>
using namespace std;
bool is_prime(int n){
    if(n==1) return false;
    else if(n==2 || n==3) return true;
    if(n%2==0) return false;
    for(int i=3;i<=int(sqrt(n));i+=2) if(n%i==0) return false;
    return true;
}
int main(){
    int n,m,cnt=0;
    cin>>n>>m;
    vector<int> num(n);
    for(int i=0;i<n;i++) cin>>num[i];
    for(int i=1;i<=m;i++){
        int t;
        cin>>t;
        if(is_prime(num[t-1])) cnt+=num[t-1];
    }cout<<cnt;
    return 0;
}