Aizu Online Judge in C++ #ITP2_8_A : Map: Search

Aizu Online Judge（AOJ）の過去問から、その提出コードの解答例です。

・問題 "Map: Search"
https://onlinejudge.u-aizu.ac.jp/problems/ITP2_8_A
キーが文字列、値が整数である辞書Mに対して、以下の操作を行ってください。ただし、辞書Mはキーの重複を許しません。
insert(key, x):にキーがkeyで値がxである要素を挿入する。キーがkeyである要素が既に存在する場合は値をxに更新する。
get(key): キーがkeyである値を出力する。

僕が作成、提出したコードは、以下のとおりです。

Aizu Online Judge in C++ #ITP2_8_A : Map: Search

/*
Aizu Online Judge in C++ #ITP2_8_A : Map: Search
https://onlinejudge.u-aizu.ac.jp/problems/ITP2_8_A
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
    map<string,int> m;
    int q;
    cin>>q;
    vector<int> n(q);
    vector<string> key(q);
    for(int i=0;i<q;i++){
        cin>>n[i]>>key[i];
        if(n[i]==0){
            int x;
            cin>>x;
            if(m.count(key[i])==1){
                m.at(key[i])=x;
            }else{
                m.insert(make_pair(key[i],x));
            }
        }else{
            cout<<m.at(key[i])<<endl;
        }
    }
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge（AOJ）」です。
http://judge.u-aizu.ac.jp/onlinejudge/