PAT乙级1038 统计同成绩学生

#include <iostream>
#include <map>
#include <vector>
using namespace std;

int main(){
int N,K,num;
vector<int> find;
map<int,int> stu_score;
cin>>N;
for(int i=0;i<N;i++){
cin>>num;
if(stu_score.count(num)==0){
stu_score[num]=1;
}else{
stu_score[num]++;
}
}
cin>>K;
for(int i=0;i<K;i++){
cin>>num;
find.push_back(num);
}
for(int i=0;i<K;i++){
if(i==0){
cout<<stu_score[find[i]];
}else{
cout<<" "<<stu_score[find[i]];
}
}
return 0;
}