[코드트리] Hash Set 연습문제

문제 : 데이터 비교

정수로 이루어진 두 수열을 비교하여 같은 원소를 가지고 있는지 판단하는 프로그램을 작성해 보세요.

---

입력 형식

첫 번째 줄에는 수열 1의 원소의 개수 n이 주어집니다.

두 번째 줄에는 수열 1의 원소들이 공백을 두고 주어집니다.

세 번째 줄에는 수열 2의 원소의 개수 m이 주어집니다.

네 번째 줄에는 수열 2의 원소들이 공백을 두고 주어집니다.

• 1 ≤ n, m ≤ 100,000
• −109 ≤ 주어지는 숫자의 범위 ≤ 109

출력 형식

수열 2의 원소의 순서대로 그 원소가 수열 1에 존재하는 원소이면 1을, 수열 1에 존재하지 않는 원소이면 0을 공백을 사이에 두고 출력합니다.

 

나의 풀이

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

int main() {
    unordered_set<int> s;
    int n; int m; int k;
    cin >> n;
    for(int i=0; i<n; ++i){
        cin >> k;
        s.insert(k);
    }
    cin >> m;
    for(int i=0; i<m; ++i){
        cin >> k;
        if(s.find(k) != s.end()) cout << 1 << " ";
        else cout << 0 << " ";
    }
    return 0;
}

 

문제 출처 코드트리