BOJ - 1920 - 수찾기

ggh·2022년 5월 11일

binary_search boj cpp ps

백준

목록 보기

18/112

BOJ - 1920 - 수찾기

문제

1920번: 수 찾기

문제 개념

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다.
다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다.
다음 줄에는 M개의 수들이 주어지는데, 이 수들이 A안에 존재하는지 알아내면 된다.
- 즉, M의 요소를 리스트 A에 있는지 유무를 판단하는 것
모든 정수의 범위는 -2^31 보다 크거나 같고 2^31 보다 작다.
- int로 받는것이 아닌 string으로 받아야 된다.

구현

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
using namespace std;

/* 
N개의 정수 A[1], A[2], …, A[N]이 주어져 있을 때,
이 안에 X라는 정수가 존재하는지 알아내는 프로그램을 작성하시오.
단
모든 정수의 범위는 -231 보다 크거나 같고 231보다 작다. 
    int로 받는것이 아닌 string으로 받도록 하자.
*/

int main()
{
    // 존재 기준 리스트
    
    // 확인할 리스트
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int A_num, M_num;
    cin >> A_num;
    vector<string> A(A_num);
    for(int i=0; i < A_num; i++)
        cin >> A[i];
    
    // 기준 리스트 오름차순 정렬 
    sort(A.begin(), A.end());

    cin >> M_num;
    vector<int> answer;
    for(int i=0; i < M_num; i++)
    {
        string temp;
        cin >> temp;
        // 이진탐색을 통한 존재유무 탐색 때문에 오름차순 정렬이 필요.
        int index = lower_bound(A.begin(), A.end(), temp) - A.begin();
				// 요소가 존재하면
        if(temp == A[index])
            answer.push_back(1);
        else
            answer.push_back(0);
    }
    
    for(auto el : answer)
        cout << el << '\n';
    
    return 0;
}