[C++] 백준 10816 숫자 카드2

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main() 
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    vector<int> table;
    int N, M, num;

    cin >> N;

    for (int i = 0; i < N; i++)
    {
        cin >> num;
        table.push_back(num);
    }

    sort(table.begin(), table.end());

    cin >> M;

    for (int i = 0; i < M; i++)
    {
        cin >> num;
        cout << upper_bound(table.begin(), table.end(), num) - lower_bound(table.begin(), table.end(), num) << " ";
    }

    return 0;
}

 

풀이

1. vector에 저장하여 정렬해준다.

2. 해당 수가 있는 처음과 끝의 차를 계산해서 몇개가 있는지를 확인한다.

3. 아래의 입출력 최적화를 빼면 시간초과난다.

ios_base::sync_with_stdio(0);
cin.tie(0);

 

느낀점

입출력 최적화가 필요할 때는 꼭 넣어야겠다.

 

https://www.acmicpc.net/problem/10816