【PAT乙级】1061 判断题

题目链接：1061 判断题

#include<iostream>
using namespace std;

int main(){
    int N, M, c, d;
    cin >> N >> M;
    int a[M],b[M];
    for(int i=0;i<M;i++) cin >> a[i];
    for(int i=0;i<M;i++) cin >> b[i];
    for(int i=0;i<N;i++){
        c = 0;
        for(int j=0;j<M;j++){
            cin >> d;
            if(d == b[j]) c += a[j];
        }
        cout << c;
        if(i != N-1) cout << endl;
    }
    return 0;
}