[백준] 이분 탐색 10815번: 숫자 카드

문제

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 가지고 있는지 아닌지를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 두 숫자 카드에 같은 수가 적혀있는 경우는 없다.

셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 가지고 있는 숫자 카드인지 아닌지를 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다

출력

첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 가지고 있으면 1을, 아니면 0을 공백으로 구분해 출력한다.

Approach

• binary search 함수를 구현하고 그 함수를 이용해 상근이가 가지고 있는 숫자카드 배열에서 입력으로 주어진 각 수를 탐색한다. 탐색해서 찾으면 1, 아니면 0을 출력한다.

1. binary search 함수 재귀로 구현
2. 상근이가 가지고 있는 숫자 카드 입력받아서 배열에 저장
3. 체크해야 할 숫자 카드 입력받아서 다른 배열에 저장
4. 상근이가 가지고 있는 숫자 카드 배열을 정렬 (for binary search)
5. 체크해야 할 숫자 카드를 하나씩 확인하면서 상근이가 가지고 있는 숫자 카드 배열에서 탐색
6. 탐색해서 찾으면 1, 아닐 경우 0을 출력

Source Code

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

// binary search 재귀 함수 구현 
int binaryRecursive(vector<int> & arr, int target, int left, int right)
{
    if (left > right)
        return -1;
    int mid = (left + right) / 2;
    
    if (arr[mid] == target)
        return mid;
    else if (arr[mid] < target)
        return binaryRecursive(arr, target, mid + 1, right);
    else
        return binaryRecursive(arr, target, left, mid - 1);
}


int main() {
    ios_base :: sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    int n; // 상근이가 가지고있는 숫자 카드 n 입력 받기 
    cin >> n;
    
    vector<int> mine; // 상근이의 숫자 카드들을 배열에 저장 
    for (int i = 0; i < n; i++)
    {
        int num;
        cin >> num;
        mine.push_back(num);
    }
    
    int m; // 체크할 카드 갯수 m 입력 받기 
    cin >> m;
    
    vector<int> cards; // 체크할 카드들을 배열에 저장 
    for (int j = 0; j < m; j++)
    {
        int num;
        cin >> num;
        cards.push_back(num);
    }
    
    sort(mine.begin(), mine.end()); // 탐색할 배열 정렬 
    
    for (int k = 0; k < m; k++) // 체크해야 할 카드들을 하나씩 확인 
    {
        if (binaryRecursive(mine, cards[k], 0, n - 1) != -1) // 찾으면 1, 아니면 0 출력
        {
            cout << 1 << " ";
        }
        else
        {
            cout << 0 << " ";
        }
    }

    return 0;
}