帮 变鸽的一只呱 改的代码

#include<iostream>
#include<cmath>
using namespace std;

bool is_prime(int n){
    if(n<2) return false;
    for(int i=2;i<=int(sqrt(n));i++) if(n%i==0) return false;
    return true;
}

int main(){
    int n, m;
    cin >> n >> m;
    int a[n];
    for(int i=0;i<n;i++) cin >> a[i];
    int sum = 0;
    for(int i=0;i<m;i++){
        int fz;
        cin >> fz;
        if(is_prime(a[fz - 1])) sum += a[fz - 1];
    }
    cout << sum << endl;
    return 0;
}