Algorithm

[코드트리] 숫자 빠르게 찾기

kwon-record 2024. 12. 20. 11:23

https://www.codetree.ai/missions/8/problems/find-number-fast/description

 

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

국가대표가 만든 코딩 공부의 가이드북 코딩 왕초보부터 꿈의 직장 코테 합격까지, 국가대표가 엄선한 커리큘럼으로 준비해보세요.

www.codetree.ai

 

 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#define endl '\n'

int binary_search(vector<int>& arr, int value){
    int left=0;
    int right=arr.size()-1;

    while(left<=right){
        int mid=(left+right)/2;
        if(value<arr[mid]){
            right=mid-1;
        }else if(value>arr[mid]){
            left=mid+1;
        }else{
            return mid+1;
        }
    }
    return -1;
}

int main() {
    int n, m;
    cin >> n >> m;

    vector<int> v(n);
    for(int i = 0; i < n; i++) {
        cin >> v[i];
    }


    for(int i = 0; i < m; i++) {
        int target;
        cin >> target;
        cout<<binary_search(v, target)<<endl;
    }

    return 0;
}