(C++) 백준 10816번 - 숫자 카드 2

문제 링크 : https://www.acmicpc.net/problem/10816

10816번: 숫자 카드 2

풀이

sort() 메서드와 해당 값보다 큰 값이 처음으로 나타는 index를 반환하는 upper_bound() 메서드, 해당 값이 처음으로 나타는 index를 반환하는 lower_bound() 메서드를 사용하면 간단하게 풀 수 있는 문제이다.

#include <iostream>
#include <algorithm>

using namespace std;

int A[500005];
int n, m;

int main() {
    ios::sync_with_stdio(0);  
    cin.tie(0);
    cout.tie(0);

    cin >> n;
    for(int i = 0; i < n; i++){
        cin >> A[i];
    }
    sort(A, A+n);

    cin >> m;
    int tmp;
    while(m--){
        cin >> tmp;
        if(!binary_search(A, A+n, tmp)){
            cout << 0 << "\n";
            continue;
        }
        cout << upper_bound(A, A+n, tmp) - lower_bound(A, A+n, tmp) << "\n";
    }
}

 

직접 upper_bound와 lower_bound를 직접 구현해서 풀이해 보면 다음과 같다.

#include <iostream>
#include <algorithm>

using namespace std;

int A[500005];
int n, m;

int lower_idx(int tgt){
    int st = 0;
    int ed = n;
    while(st < ed){
        int mid = (st+ed)/2;
        if(A[mid] >= tgt) ed = mid;
        else st = mid+1;
    }
    return st;
}

int upper_idx(int tgt){
    int st = 0;
    int ed = n;
    while(st < ed){
        int mid = (st+ed)/2;
        if(A[mid] > tgt)     ed = mid;
        else st = mid+1;
    }
    return st;
}

int main() {
    ios::sync_with_stdio(0);  
    cin.tie(0);
    cout.tie(0);

    cin >> n;
    for(int i = 0; i < n; i++){
        cin >> A[i];
    }
    sort(A, A+n);

    cin >> m;
    int tmp;
    while(m--){
        cin >> tmp;
        if(!binary_search(A, A+n, tmp)){
            cout << 0 << " ";
            continue;
        }
        cout << upper_bound(A, A+n, tmp) - lower_bound(A, A+n, tmp) << " ";
    }
}