BOJ - 10815 - 숫자 카드
문제
문제 개념
문제
- 숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 가지고 있는지 아닌지를 구하는 프로그램을 작성하시오.
입력
- 첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 두 숫자 카드에 같은 수가 적혀있는 경우는 없다.
- 수가 너무 크기에 string으로 받도록 한다.
셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 가지고 있는 숫자 카드인지 아닌지를 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다
이전 포스팅에서 풀어본 수 찾기 문제와 매우 유사하다.
구현
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
using namespace std;
/*
N개의 정수 A[1], A[2], …, A[N]이 주어져 있을 때,
이 안에 X라는 정수가 존재하는지 알아내는 프로그램을 작성하시오.
단
모든 정수의 범위는 -231 보다 크거나 같고 231보다 작다.
int로 받는것이 아닌 string으로 받도록 하자.
*/
int main()
{
// 존재 기준 리스트
// 확인할 리스트
ios::sync_with_stdio(false);
cin.tie(nullptr);
int A_num, M_num;
cin >> A_num;
vector<string> A(A_num);
for(int i=0; i < A_num; i++)
cin >> A[i];
// 기준 리스트 오름차순 정렬
sort(A.begin(), A.end());
cin >> M_num;
vector<int> answer;
for(int i=0; i < M_num; i++)
{
string temp;
cin >> temp;
// 이진탐색을 통한 존재유무 탐색 때문에 오름차순 정렬이 필요.
int index = lower_bound(A.begin(), A.end(), temp) - A.begin();
// 요소가 존재하면
if(temp == A[index])
answer.push_back(1);
else
answer.push_back(0);
}
for(auto el : answer)
cout << el << '\n';
return 0;
}
SOL
- string으로 요소를 받는다.
- 비교할 땐 이진탐색 기반인 lower_bound를 사용
- lower_bound를 사용하기 위해 기준 리스트 오름차순 정렬