寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #ITP2_8_B : Map: Delete

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

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

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

Aizu Online Judge in C++ #ITP2_8_B : Map: Delete
/*
Aizu Online Judge in C++ #ITP2_8_B : Map: Delete
https://onlinejudge.u-aizu.ac.jp/problems/ITP2_8_B
 提出コードの解答例
 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 if(n[i]==1){
            if(m.count(key[i])==0){
                cout<<0<<endl;
            }else{
                cout<<m.at(key[i])<<endl;
            }
        }else{
            m.erase(key[i]);
        }
    }
    return 0;
}

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