[코드트리] TreeMap 연습문제

문제

n개의 명령이 주어졌을 때, 각 명령을 수행하는 프로그램을 작성해보세요. 명령의 종류는 크게 4가지 입니다.

• add k v : (k, v) 쌍을 treemap에 추가합니다. key가 k, value가 v라는 뜻입니다. 이때 만약 동일한 k가 이미 존재한다면, v로 덮어씁니다.
• remove k : key가 k인 쌍을 찾아 treemap에서 제거합니다. 잘못된 입력은 주어지지 않습니다.
• find k : key가 k인 쌍이 treemap에 있는지를 판단합니다. 있다면 해당하는 value를 출력하고, 없다면 None을 출력합니다.
• print_list : treemap에 있는 쌍들을 key 기준으로 오름차순 정렬하여 각 value 값들만 공백을 사이에 두고 순서대로 출력합니다. 만약 treemap이 비어있다면 None을 출력합니다.

---

입력 형식

첫 번째 줄에는 n이 주어집니다.

두 번째 줄 부터는 n개의 줄에 걸쳐 각 명령이 한 줄에 하나씩 주어집니다. 각 명령에 주어지는 key와 value는 전부 숫자입니다. 명령들은 순서대로 수행되어야 합니다. 단, print_list 명령은 최대 10번만 주어진다고 가정해도 좋습니다.

• 1 ≤ n ≤ 100,000
• 1 ≤ 주어지는 숫자 ≤ 109

출력 형식

결과를 한 줄에 하나씩 출력합니다.

 

나의 풀이

 

#include <iostream>
#include <string>
#include <map>

using namespace std;

int n;
map<int, int> m;

int main() {
    cin >> n;
    for(int i = 0; i < n; i++) {
        string command;
        cin >> command;

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

 

 

문제 출처 코드트리