C++ 백준 10816번

백준 10816번 www.acmicpc.net/problem/10816

10816번: 숫자 카드 2

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

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

 

개수 제한이 50만개라 하나하나 비교하면 시간이 너무 오래 걸린다. 더 빠른 방법을 생각해보자!!

 

상한(upper bound)은 큰 수 중 첫 번째 숫자이고

하한(lower bound)은 크거나 같은 수 중에서 첫 번째 숫자를 뜻한다.

 

그러니까 어떤 배열이 오름차순으로 정렬되어있다고 생각해보면

1 1 2 2 4 4 4 4 5 5 6 6 6 6

여기서 4의 상한을 찾아보면 상한은 큰 수 중 첫번째 숫자니까

1 1 2 2 4 4 4 4 5 5 6 6 6 6 이 첫번째 5이고

4의 하한을 찾아보면 하한은 크거나 같은 수 중 첫 번째 숫자니까

1 1 2 2 4 4 4 4 5 5 6 6 6 6 이 첫번째 4이다.

그래서 이 5(4의 상한)의 인덱스는 8이고 4(4의 하한)의 인덱스는 4니까 상한의 인덱스에서 하한의 인덱스를 빼면

총 4의 개수인 4개가 나온다!!!

 

이걸 이용해서 코드를 짜면 된다. 상한과 하한의 인덱스는 algorithm에 있는 lower_bound와 upper_bound를 사용하면 된다!!

#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n;
	cin >> n;
	vector<int> a(n);
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	sort(a.begin(), a.end());
	
	int m;
	cin >> m;
	for (int i = 0; i < m; i++) {
		int number;
		cin >> number;
		auto l = lower_bound(a.begin(), a.end(), number); // 하한 : 같거나 큰거 중에 첫번째 index
		auto r = upper_bound(a.begin(), a.end(), number); // 상한 : 큰거 중에 첫번째 index
		cout << r - l << ' '; // 상한 - 하한 하면 총 같은 개수 나옴
	}
	cout << '\n';
}