[백준] 2776 암기왕

2776번: 암기왕

 

www.acmicpc.net

풀이

문제 분류를 보면 이분 탐색을 의도한 것 같지만, HashSet을 사용해서 해결했다. C++에서는 HashSet이 unordered_set이므로 해당 STL을 사용해서 구현했다.

코드

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

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int t;
    cin >> t;
    int n, m;
    unordered_set<int> s;

    while (t--) {
        cin >> n;
        s.clear();
        for (int i = 0; i < n; i++) {
            int tmp;
            cin >> tmp;
            s.insert(tmp);
        }
        cin >> m;
        for (int i = 0; i < m; i++) {
            int tmp;
            cin >> tmp;
            if (s.find(tmp) != s.end()) cout << "1\n";
            else cout << "0\n";
        }
    }
    return 0;
}