백준 1620 포켓몬 마스터 c++

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

 

처음에는 그냥 넣고 선형탐색으로 해서 시간초가 나왔는데 해시맵으로 index번호와 함께 저장하면 바로 찾을 수 있어서 탐색이 필요 없다.

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<string>

using namespace std;

vector <vector<int>> v;

int main() {

	int n, m;
	cin >> n >> m;

	vector<string> v1(n);
	vector<int> v2(m);

	for (int i = 0; i < n; i++) {
		
		cin >> v1[i];

	}

	for (int i = 0; i < m; i++) {
		string temp;
		cin >> temp;

		if (temp[0] < 64) {
			int temp1 = stoi(temp);
			cout << v1[temp1] << '\n';
		}

		else {
			for (int k = 0; k < n; k++) {
				if (v1[k] == temp)
					cout << k+1 << '\n';
			}
		}

	}

	return 0;
}