C++のmap・unordered_mapでキーが存在するか確認する方法

C++のstd::mapやstd::unordered_mapは、キーとそれに対応する値をペアで管理する連想コンテナです。本記事では、指定したキーがマップ内に存在するかどうかを判定する方法を解説します。

find()を使ったキーの存在チェック

最も一般的な方法は、find()メンバ関数を使うことです。find()は指定したキーを検索し、見つかった場合はその要素へのイテレータを返します。キーが存在しない場合は、終端イテレータend()が返されるため、これを利用して存在判定を行えます。

サンプルコード

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

string isPresent(map<string, int> m, string key) {
    if (m.find(key) == m.end())
        return "Not Present";
    return "Present";
}

int main() {
    map<string, int> my_map;
    my_map["first"] = 4;
    my_map["second"] = 6;
    my_map["third"] = 6;

    string check1 = "fifth", check2 = "third";
    cout << check1 << ": " << isPresent(my_map, check1) << endl;
    cout << check2 << ": " << isPresent(my_map, check2);
}

実行結果

fifth: Not Present
third: Present

この例では、「fifth」というキーはマップに登録されていないため「Not Present」が、「third」は登録済みのため「Present」が出力されます。

count()を使う代替手段

count()メンバ関数を使っても同様の判定が可能です。マップでは各キーは最大1つしか存在しないため、戻り値が1ならキーが存在し、0なら存在しないことを意味します。

if (my_map.count("third") > 0) {
    // キーが存在する場合の処理
}

C++20以降ならcontains()が便利

C++20からは、より直感的なcontains()メンバ関数が追加されました。キーが存在すればtrue、しなければfalseを返すため、コードの意図が明確になります。

if (my_map.contains("third")) {
    // キーが存在する場合の処理
}

mapとunordered_mapの違い

std::mapは内部に赤黒木（平衡二分探索木）を使用しており、キーが自動的にソートされた状態で保持されます。一方、std::unordered_mapはハッシュテーブルをベースとしており、平均的に高速な検索が可能です。どちらのコンテナでも、上記のfind()・count()・contains()によるキー存在チェックは同じように動作します。