17219번: 비밀번호 찾기

문제링크

C++

#include <iostream>
#include <map>

using namespace std;

int N, M;

map<string, string> m1;

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

    cin >> N >> M;
    for (int i = 0; i < N; i++){
        string tmp1, tmp2;
        cin >> tmp1 >> tmp2;
        m1.insert(pair<string, string>(tmp1, tmp2));

    }
    for (int i = 0; i < M; i++){
        string tmp;
        cin >> tmp;
        cout << m1.find(tmp)->second << "\n";
    }
}

Python

import sys

diction = {}

N, M = map(int, input().split())

for _ in range(N):
    tmp = sys.stdin.readline().split()
    diction[tmp[0]] = tmp[1]

for _ in range(M):
    print(diction[input()])

• cpp는 맵, 파이썬은 딕셔너리 사용
• 맵은 $O ( max ( N \cdot log(N) , M \cdot log(M) ))$
• 딕셔너리는 해시이므로 $O ( max( N, M))$