알고리즘/문제 풀이

1920번: 수 찾기

Themion 2021. 12. 10. 15:52

https://www.acmicpc.net/problem/1920

 

1920번: 수 찾기

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들

www.acmicpc.net

배열의 길이와 M이 각각 100,000이므로, 배열을 정렬한 뒤 각각의 수에 대해 이분 탐색을 실행하면 된다.

#include <algorithm>
#include <iostream>

using namespace std;

#define MAX_N 100000

int main() {
    // 입출력 속도 향상
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    // N: 정수의 개수, M: 찾을 수의 개수, A: 정수를 저장할 배열
    // buf: 찾을 수를 저장할 공간
    int N, M, A[MAX_N], buf;

    //배열의 크기와 해당 크기만큼의 배열을 입력받은 뒤 정렬한다
    cin >> N;
    for (int i = 0; i < N; i++) cin >> A[i];
    sort(A, A + N);

    //찾을 수의 개수를 입력받은 뒤
    cin >> M;
    while (M--) {
        //해당 횟수만큼 수를 입력받아 이진탐색을 실행한다
        cin >> buf;
        cout << binary_search(A, A + N, buf) << '\n';
    }

    return 0;
}