BOJ : 2276 암기왕

문제 : https://www.acmicpc.net/problem/2776

github : https://github.com/JEONG-SEUNGWOOK/BOJ/blob/master/2276.cpp

기초적인 이진탐색 문제.

테스트 케이스도 하나가 아니기 때문에 반복문을 이용하면 TLE.

algorithm 라이브러리에 binary_search 함수가 있기 때문에 사용하면 더 편하게 구현 가능하다.

#include <cstdio>
#include <algorithm>
using namespace std;
bool bs(int size,int a[],int m){
    int low=0, high=size-1, mid;
    while(low<=high){
        mid = low+(high-low)/2;
        if(a[mid]<m) low=mid+1;
        else if(a[mid]>m)high=mid-1;
        else return true;
    }
    return false;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int N,M;
        int a[1000001];
        scanf("%d",&N);
        for(int i=0;i<N;i++) scanf("%d",&a[i]);
        sort(a,a+N);
        scanf("%d",&M);
        while(M--){
            int t;
            scanf("%d",&t);
            printf("%d\n",bs(N,a,t));
        }
    }
}