[이분탐색] 백준 10816 :: 숫자 카드 2

숫자 카드 2 성공

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
1 초	256 MB	9894	2933	2017	33.723%

문제

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 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 

10
6 3 2 10 10 10 -10 -10 7 3
8
10 9 -5 2 3 4 5 -10

예제 출력 1 

3 0 0 1 2 0 0 2

     

 문제 풀이

1차 풀이 (  3940 KB,  348 ms)

기본적인 이분탐색 문제.

lower_bound와 upper_bound를 구하는 문제이다.

찾은 lower_bound가 실제 그 값인지 체크해야하고,

ans가 한번도 갱신되지 않은 경우를 고려하여, 기본값을 N으로 해야 한다.

#include <bits/stdc++.h>
using namespace std;
using int64 = int64_t;

int N, num[500'000];
int M;

bool lower_bound_test(const int& x, const int& val){
        return val <= num[x];
}
bool upper_bound_test(const int& x, const int& val){
        return val < num[x];
}

int main(){
        ios_base::sync_with_stdio(false);
        cin.tie(nullptr);


        //! data input.
        cin >> N;
        for(auto i=0; i<N; i++){
                cin >> num[i];
        }
        sort(num, num+N);
        cin >> M;


        //! soving.
        int ret_srt, ret_end;
        int srt, end, mid, ans, find_to;
        while(M--){
                //! get find_to
                cin >> find_to;


                //! lower_bound
                srt = 0;
                end = N;
                ans = N;
                while(srt < end){
                        mid = (srt + end)/2;
                        if(lower_bound_test(mid, find_to)){
                                ans = mid;
                                end = mid;
                        }
                        else srt = mid+1;
                }
                ret_srt = ans;


                //! upper_bound
                srt = 0;
                end = N;
                ans = N;
                while(srt < end){
                        mid = (srt + end)/2;
                        if(upper_bound_test(mid, find_to)){
                                ans = mid;
                                end = mid;
                        }
                        else srt = mid+1;
                }
                ret_end = ans;


                //! check vaild.;
                if(ret_srt < 0) cout << 0 << ' ';
                else if(num[ret_srt] != find_to) cout << 0 << ' ';
                else cout << ret_end - ret_srt << ' ';
        }
}