Решение задачи Баллы с Acmp

#include <bits/stdc++.h>
#include <stdio.h>
using namespace std;
 
int main()
{
    int n,m;
    scanf("%d %d", &n, &m);
    std::set<long long> set;
    for(int i = 0; i < n; ++i){
        long long x;
        scanf("%lld", &x);
        set.insert(x);
    }
    for(int i = 0; i < m; ++i){
        long long x;
        scanf("%lld", &x);
        if(i > 0)
            printf(" ");
        if(set.count(x) > 0)
            printf("YES");
        else
             printf("NO");
    }
    return 0;
}