BOJ - 1920 - 수찾기
문제
문제 개념
- 첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다.
- 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다.
- 다음 줄에는 M개의 수들이 주어지는데, 이 수들이 A안에 존재하는지 알아내면 된다.
- 즉, M의 요소를 리스트 A에 있는지 유무를 판단하는 것
- 모든 정수의 범위는 -2^31 보다 크거나 같고 2^31 보다 작다.
- int로 받는것이 아닌 string으로 받아야 된다.
구현
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를 사용하기 위해 기준 리스트 오름차순 정렬