C++ map, set 에서 중복 키 삽입을 확인하는 방법

C++에서는 insert 함수의 리턴값을 활용할 수 있다.

insert 함수의 프로토타입: pair<iterator,bool> insert (const pair<const Key, Type>& val);

 

중복된 Key가 존재하지 않아 삽입 성공 시, <insert되어진 element의 iterator, true> 의 pair를 반환한다.
중복된 Key가 존재해, 삽입 실패 시, <중복된 Key의 element의 iterator, false> 의 pair를 반환한다.

#include <iostream>
#include <map>
using namespace std;

int main() {
    int T, N, i, j, ans;
    string value, key;
    map<string, int> m;
    pair<map<string, int>::iterator, bool> pr;

    cin >> T;
    while (T-- > 0) {
        m.clear();
        ans = 1;
        cin >> N;
        for (j = 0; j < N; j++) {
            cin >> value >> key;
            pr = m.insert(pair<string, int>(key, 1)); // 주목 !
            if (!pr.second) m[key]++; // Key가 중복되어 map에 삽입하지 못한 경우엔 직접 1을 더해줌.
        }
        for (map<string, int>::iterator it = m.begin(); it != m.end(); it++) {
            ans *= it->second + 1;
        }
        cout << ans - 1 << "\n";
    }
}

boj 9375 패션왕 신혜빈 문제를 풀다가 새로 알게 된 정보이다.

map, set 을 활용하다보면 문제 구조 상 중복된 key를 일부러 삽입해보고 확인하는 경우가 있다.

이 문제의 경우, 중복된 Key 값이 몇 번 입력되었는지 횟수를 저장해 문제를 풀어야 한다.

 

std::map<std::string, int> temp;
auto itr = temp.find("my_key");
if (itr == temp.end()) {
    temp.insert({ "my_key", 1 });
} else {
    itr->second++;
}

이렇게 잘못된 코드로 코딩하는 사람들이 꽤 많은데 올바르지 않은 방법이다!

 

참고 링크: https://yonmy.com/archives/9