https://www.acmicpc.net/problem/28446
동현이의 볼링공은 무게가 모두 다르기 때문에 2번 요청의 답은 항상 하나만 존재한다.
같은 무게에 대한 번호를 저장할 필요는 없습니다.
무게를 입력하면 번호를 반환받을 수 있게 구현해 주면 됩니다.
#include <iostream>
#include <unordered_map>
using namespace std;
int M, cmd, w, x;
unordered_map<int, int> lockers;
int main()
{
ios::sync_with_stdio(0), cin.tie(0);
cin >> M;
while (M--)
{
cin >> cmd;
switch (cmd)
{
case 1:
cin >> x >> w;
lockers[w] = x;
break;
case 2:
cin >> w;
cout << lockers[w] << "\n";
break;
}
}
return 0;
}
무게를 키, 번호를 값으로 활용하는 맵을 활용하면 됩니다.