[코드트리 챌린지] 8주차 - TreeMap / treemap 기본

 

https://www.codetree.ai/missions/8/problems/treemap-basic/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

코드:

#include <bits/stdc++.h>
#include <map>
using namespace std;

int n;
map<int, int> m;

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

    string order;
    int k, v;

    while (n--) {
        cin >> order;

        if (order == "add") {
            cin >> k >> v;
            m[k] = v;
        }
        else if (order == "remove") {
            cin >> k;
            m.erase(k);
        }
        else if (order == "find") {
            cin >> k;
            if (m.find(k) != m.end())
                cout << m[k];
            else
                cout << "None";
            cout << "\n";
        }
        else {
            if (m.size() == 0)
                cout << "None";
            else {
                map<int, int>::iterator it;
                for (it = m.begin(); it != m.end(); it++)
                    cout << it->second << " ";
            }
            cout << "\n";

        }

       
    }
}

map의 경우 key에 대해 오름차순 정렬이 되어 있으므로 iterator를 통해 순서대로 출력이 가능합니다