如何在调用方判断map中键是否存在并访问返回的iterator

方案1：给类A添加end()接口 #

直接在类A中暴露返回底层map的end()迭代器的方法，让调用方可以像操作普通map一样判断迭代器有效性：

class A
{
    std::unordered_map<int,int> m = { {5,100} };

public:
    std::unordered_map<int,int>::iterator get(int key)
    {
        return m.find(key);
    }  

    // 添加该方法，返回底层map的end迭代器
    std::unordered_map<int,int>::iterator end()
    {
        return m.end();
    }
};

int main()
{
    A a;
    
    auto keyIter = a.get(5);
    if (keyIter != a.end()) {
        // 元素存在，可访问keyIter->first/second
        std::cout << "值为：" << keyIter->second << std::endl;
    }
}

这个方案最贴近原生map的使用逻辑，无需改动原有get()方法，仅补充一个简单接口即可。

方案2：返回包含存在性标记的pair #

让get()方法返回一个std::pair，第一个值表示元素是否存在，第二个值是迭代器，避免调用方依赖底层map的end()：

#include <utility>

class A
{
    std::unordered_map<int,int> m = { {5,100} };

public:
    std::pair<bool, std::unordered_map<int,int>::iterator> get(int key)
    {
        auto it = m.find(key);
        return {it != m.end(), it};
    }
};

int main()
{
    A a;
    
    // C++17可用结构化绑定，旧版本用pair.first/pair.second
    auto [exists, keyIter] = a.get(5);
    if (exists) {
        std::cout << "值为：" << keyIter->second << std::endl;
    }
}

方案3：返回std::optional（C++17及以上） #

用std::optional包装迭代器，元素不存在时返回空的optional，更符合现代C++风格：

#include <optional>

class A
{
    std::unordered_map<int,int> m = { {5,100} };

public:
    std::optional<std::unordered_map<int,int>::iterator> get(int key)
    {
        auto it = m.find(key);
        return it != m.end() ? std::optional(it) : std::nullopt;
    }
};

int main()
{
    A a;
    
    auto keyOpt = a.get(5);
    if (keyOpt.has_value()) {
        auto keyIter = *keyOpt;
        std::cout << "值为：" << keyIter->second << std::endl;
    }
}

如果想直接返回值的引用，也可以用std::optional<std::reference_wrapper<int>>，调用方无需处理迭代器：

#include <optional>
#include <functional>

class A
{
    std::unordered_map<int,int> m = { {5,100} };

public:
    std::optional<std::reference_wrapper<int>> get(int key)
    {
        auto it = m.find(key);
        return it != m.end() ? std::ref(it->second) : std::nullopt;
    }
};

int main()
{
    A a;
    
    auto valOpt = a.get(5);
    if (valOpt) {
        int& val = valOpt.value();
        std::cout << "值为：" << val << std::endl;
        // 还能直接修改值
        val = 200;
    }
}

方案4：拆分存在性检查与取值接口 #

把存在性判断和取值拆成两个独立方法，逻辑更清晰：

class A
{
    std::unordered_map<int,int> m = { {5,100} };

public:
    // 仅检查元素是否存在
    bool contains(int key) const
    {
        return m.find(key) != m.end();
    }

    // 取值，调用方需先通过contains确认元素存在，否则at()会抛出std::out_of_range异常
    int& get(int key)
    {
        return m.at(key);
    }
};

int main()
{
    A a;
    
    if (a.contains(5)) {
        int& val = a.get(5);
        std::cout << "值为：" << val << std::endl;
    }
}

内容的提问来源于stack exchange，提问作者xyf