Home BOJ - 1920 - 수찾기
Post
Cancel

BOJ - 1920 - 수찾기

BOJ - 1920 - 수찾기

문제


1920번: 수 찾기

https://user-images.githubusercontent.com/71277820/165308600-4e5434eb-d807-4d96-b4ef-96f68f24c74c.png

문제 개념


  • 첫째 줄에 자연수 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를 사용하기 위해 기준 리스트 오름차순 정렬

Reference & 다른 PS 모음집


https://github.com/ggh-png/PS

C++ STL lower_bound ( )

This post is licensed under CC BY 4.0 by the author.