A296.体育课题解

#include<iostream>
#include<cmath>
using namespace std;
int x(int x){
    unsigned int i;
    if(x==1){
        return 0;
    }
    if(x==2){
        return 1;
    }
    for(i=2;i<ceil(x/2)+1;i++){
        if(x%i==0){
            return 0;
        }
    }
    return 1;
}
int main(){
    unsigned int n,m,i,a=0;
    int n1[2000],m1[2000];
    cin>>n>>m;
    for(i=0;i<n;i++){
        cin>>n1[i];
    }
    for(i=0;i<m;i++){
        cin>>m1[i];
        if(x(n1[m1[i]-1])==1){
            a+=n1[m1[i]-1];
        }
    }
    cout<<a;
}