Judge - 1236 : 넌 얼마나 빠르게 왔었니?

출처 : https://judge.koreatech.ac.kr/problem.php?id=1236

 

map 자료구조를 이용해서 중복된 학생의 순서들을 기록해서 풀었다.

unordered_map을 사용하니 실행속도가 더 빨랐다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int n, q;
    cin >> n;
    unordered_map<int, vector<int>> mp;
    for (int i = 1; i <= n; i++) {
        int num;
        cin >> num;
        if (mp.find(num) == mp.end()) {
            mp.insert({ num,vector<int>() });
        }
        mp[num].push_back(i);
    }
    cin >> q;
    for (int i = 0; i < q; i++) {
        int num;
        cin >> num;
        if (mp.find(num) == mp.end()) {
            cout << "-1" << "\n";
        }
        else {
            for (auto a : mp[num]) {
                cout << a << " ";
            }
            cout << "\n";
        }
    }
}