[코드트리 챌린지] 8주차 - HashSet / 데이터 비교

 

https://www.codetree.ai/missions/8/problems/data-comparison/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

코드:

#include <bits/stdc++.h>
#include <unordered_set>
using namespace std;

int n, m;
unordered_set<int> set1; // 수열1의 원소 종류 저장
unordered_set<int> set2; // 수열2의 원소 종류 저장
int arr[100005]; // 수열2의 원소 저장

int main(void) {
    ios::sync_with_stdio(0);
    cin.tie(0);
   
    cin >> n;

    int k;
    for (int i = 0; i < n; i++) {
        cin >> k;
        set1.insert(k);
    }

    cin >> m;
    for (int i = 0; i < m; i++) {
        cin >> k;
        arr[i] = k; // 나중에 원소 존재여부를 확인하기 위해 사용
        set2.insert(k);
    }


    for (int i = 0; i < m; i++) {
        // 수열2의 원소가 수열1 종류 중에 존재하는지 확인
        if (set1.find(arr[i]) != set1.end())
            cout << 1;
        else
            cout << 0;
        cout << "\n";
    }
}