[코드트리] HashMap 연습문제

Hash Map 사용 예시, 해시맵, code tree, cpp, 자료구조, 문제, 예제

 

문제

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

• add k v : (k, v) 쌍을 hashmap에 추가합니다. key가 k, value가 v라는 뜻입니다. 이때 만약 동일한 k가 이미 존재한다면, v로 덮어씁니다.
• remove k : key가 k인 쌍을 찾아 hashmap에서 제거합니다. 잘못된 입력은 주어지지 않습니다.
• find k : key가 k인 쌍이 hashmap에 있는지를 판단합니다. 있다면 해당하는 value를 출력하고, 없다면 None을 출력합니다.

---

입력 형식

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

두 번째 줄 부터는 n개의 줄에 걸쳐 각 명령이 한 줄에 하나씩 주어집니다. 각 명령에 주어지는 key와 value는 전부 숫자입니다. 명령들은 순서대로 수행되어야 합니다.

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

출력 형식

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

 

 

나의 풀이

 

#include <iostream>
#include <string>
#include <unordered_map>
using namespace std;

int main() {
    string st;
    int n;
    int k=0; int v=0;
    unordered_map<int, int> m;
    cin >> n;
    for(int i=0; i<n; ++i){
        cin >> st >> k;
       if(st=="add") {
           cin >> v;
           m[k] = v;
        }
       if(st=="remove") m.erase(k);
       if(st=="find"){
           if(m.find(k) != m.end()){
               cout << m[k]<<"\n";
           }else{
               cout<<"None"<<"\n";
           }
       }
    }
    return 0;
}

 

 

문제 출처 코드트리