백준 수찾기 [c++]

수찾기 실버4

문제

N개의 정수 A[1], A[2], …, A[N]이 주어져 있을 때, 이 안에 X라는 정수가 존재하는지 알아내는 프로그램을 작성하시오.

입력

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들이 A안에 존재하는지 알아내면 된다. 모든 정수의 범위는 -231 보다 크거나 같고 231보다 작다.

출력

M개의 줄에 답을 출력한다. 존재하면 1을, 존재하지 않으면 0을 출력한다.

풀이

• 이진탐색으로 풀었는데 시간초과났다.
• 입력때문인가 해서 endl대신 printf사용해봄. -> 시간초과
• ios_base::sync_with_studio(0)와 cin.tie(0)로 속도향상.
• ios_base::sync_with_studio(0) : 일반적인 경우 stdio와 iostream이 동기화되어 따로 버퍼를 갖는데 이를 사용하면 이 둘의 동기화를 풀어 버퍼를 따로 가지게 해, 속도향상됨.(싱글 쓰레드에서)
• cin.tie(0) : cin과 cout는 묶여 있는데 이를 풀어 속도향상.

코드

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

int arr[100001];

int bin_search(int num, int start ,int end){

    int half = (end + start) / 2;

    if(start > end){
        return 0;
    }

    if(arr[half] == num){
        return 1;
    } else if(arr[half] < num){
        return bin_search(num, half + 1, end);
    } else{
        return bin_search(num, start, half - 1);
    }

}

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n, m;
    cin >> n;
    for(int i = 0; i < n; i++)
    {
        cin >> arr[i];
    }
    sort(arr, arr+n);
    
    cin >> m;

    for(int i = 0; i < m ; i++)
    {
        int temp;
        cin >> temp;
        // cout << bin_search(temp, 0, n-1) << endl;
        printf("%d\n", bin_search(temp, 0, n-1));
    }


    return 0;
}

• ps
• 구현