[백준] 10816번: 숫자 카드 2 풀이 및 해설(C++)

 

https://www.acmicpc.net/problem/10816

10816번: 숫자 카드 2

 

C++로 작성한 백준 문제 10816번 숫자 카드 2에 대한 문제 풀이다.

실버 4 문제로 자료구조, 정렬, 이분탐색, 해싱이 문제를 푸는데 필요한 주요 지식이다.

 

 

---

문제 설명

문제

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 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개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 몇 개 가지고 있는지를 공백으로 구분해 출력한다.

 

---

문제 파악하기

 

 

해당 문제와 유사한 문제다.

2024.01.14 - [algorithm/Binary Search] - [백준] 1920번: 수 찾기 풀이 및 해설(C++)

[백준] 1920번: 수 찾기 풀이 및 해설(C++)

 

upper_bound() 함수와 lower_bound()함수에 대해 제대로 모른다면, 위의 글을 보고 오자.

 

위 문제에서 사용한 upper_bound() 함수와 lower_bound() 함수를 이용한다면 같은 방식으로 풀 수 있다.

 

 

---

 

코드 및 설명

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#define fastio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
using namespace std;


int main()
{
    fastio;
    int N,M,a;
    cin>>N;
    vector<int> v;
    for(int i=0; i<N; i++){
        cin>>a;
        v.push_back(a);
    }
    sort(v.begin(), v.end());
    cin >>M;
    for(int i=0; i<M; i++){
        cin>>a;
        cout<<upper_bound(v.begin(),v.end(),a)-lower_bound(v.begin(),v.end(),a)<<' ';
    }


    return 0;
}

 

upper_bound(시작범위, 끝범위, 찾으려는 값) - lower_bound(시작범위, 끝범위, 찾으려는 값)는

 

찾으려는 값의 개수가 몇 개 있는지를 표시해 준다.

 

 

---

따라서 그냥 두 함수를 써서 풀어주면 된다.

 

헤매었던 부분

딱히 없다.

 

 

 

 

 

---

 

마무리

upper_bound()와 lower_bound()를 이용한 발상이 중요하다.