문제
숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 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
#include <iostream>
using namespace std;
int arr1[500001];
int arr2[500001];
int temp[500001];
void MergeSort(int list[], int start, int mid, int end)
{
int i = start;
int j = mid + 1;
int c = start;
while (i <= mid && j <= end)
{
if (list[i] <= list[j])
{
temp[c++] = list[i++];
}
else
{
temp[c++] = list[j++];
}
}
while (i <= mid)
{
temp[c++] = list[i++];
}
while (j <= end)
{
temp[c++] = list[j++];
}
for (int k = start; k <= end; k++)
{
list[k] = temp[k];
}
}
void Merge(int list[], int start, int end)
{
if (start < end)
{
int mid = (start + end) / 2;
Merge(list, start, mid);
Merge(list, mid + 1, end);
MergeSort(list, start, mid, end);
}
}
int BinarySerch_LowerBound(int start, int end, int find)
{
int left = start, right = end, mid;
int lower_bnd = -1;
while (left <= right)
{
mid = (left + right) / 2;
if (arr1[mid] >= find)
{
lower_bnd = mid;
right = mid - 1;
}
else
{
left = mid + 1;
}
}
return lower_bnd;
}
int BinarySerch_UpperBound(int start, int end, int find)
{
int left = start, right = end, mid;
int upper_bnd = -1;
while (left <= right)
{
mid = (left + right) / 2;
if (arr1[mid] <= find)
{
upper_bnd = mid;
left = mid + 1;
}
else
{
right = mid - 1;
}
}
return upper_bnd + 1;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int N, N2;
cin >> N;
for (int i = 0; i < N; i++)
{
cin >> arr1[i];
}
Merge(arr1, 0, N - 1);
cin >> N2;
for (int i = 0; i < N2; i++)
{
cin >> arr2[i];
}
int position, count;
for (int i = 0; i < N2; i++)
{
if (BinarySerch_UpperBound(0, N - 1, arr2[i]) != -1 && BinarySerch_LowerBound(0, N - 1, arr2[i]) != -1)
{
cout << BinarySerch_UpperBound(0, N - 1, arr2[i]) - BinarySerch_LowerBound(0, N - 1, arr2[i]) << " ";
}
else
cout << "0 ";
}
return 0;
}
728x90
반응형
'알고리즘 > solved.ac' 카테고리의 다른 글
[class2] (백준 1874) 스택 수열 (0) | 2021.10.20 |
---|---|
[class2] (백준 11866) 요세푸스 문제 0 (0) | 2021.10.19 |
[class2] (백준 10866) 덱 (0) | 2021.10.17 |
[class2] (백준 10845) 큐 (0) | 2021.10.16 |
[class2] (백준 10828) 스택 (0) | 2021.10.15 |