如何用迭代器访问std::set元素？双向字典实现崩溃排查

核心错误原因 #

代码崩溃的根本问题是const迭代器与非const迭代器类型不兼容，导致std::map的查找操作触发未定义行为：

• 在TranslateForward和TranslateBackward这些const成员函数中，words.find()返回的是std::set<std::string>::const_iterator，但你定义的forward和backward两个map的键类型是std::set<std::string>::iterator（非const迭代器）。
• 尽管自定义了比较器通过迭代器指向的字符串内容比较，但const_iterator和iterator是不同类型，无法直接匹配map中存储的键。用const_iterator查找非const键的map，会导致无法找到目标条目，甚至触发非法内存访问（比如at()函数找不到键时抛出未捕获的异常，引发崩溃）。

修复步骤 #

1. 统一迭代器类型：将forward和backward的键值类型都改为const_iterator，无需修改std::set元素时，const迭代器更安全且符合const成员函数要求。
2. 调整自定义比较器：将比较器的参数改为const_iterator类型。
3. 修正查找逻辑：在Translate系列函数中，将迭代器变量声明为const_iterator。
4. 避免重复映射：在Add函数中先检查映射是否已存在，避免插入冗余条目（可选优化）。

修复后的完整代码 #

struct comparator {
    bool operator() (const std::set<std::string>::const_iterator& lhs, const std::set<std::string>::const_iterator& rhs) const {
        return *lhs < *rhs;
    }
};

class Translator {
public:
    Translator() = default;
    void Add(std::string_view source, std::string_view target) {        
        auto [fw, fw_ok] = words.insert(std::string(source));
        auto [bw, bw_ok] = words.insert(std::string(target));
        const auto fwc = fw;
        const auto bwc = bw;
        
        // 避免重复插入相同映射
        if (forward.find(fwc) == forward.end()) {
            forward.insert({fwc, bwc});
        }
        if (backward.find(bwc) == backward.end()) {
            backward.insert({bwc, fwc});
        }
    }
    
    std::string_view TranslateForward(std::string_view source) const {
        std::set<std::string>::const_iterator it = words.find(std::string(source));
        if (it == words.end() || forward.count(it) == 0) {
            return {};
        }
        return *(forward.at(it));             
    }
    
    std::string_view TranslateBackward(std::string_view target) const {
        std::set<std::string>::const_iterator it = words.find(std::string(target));
        if (it == words.end() || backward.count(it) == 0) {
            return {};
        }
        return *(backward.at(it));             
    }

private:
    std::set<std::string> words;
    std::map<std::set<std::string>::const_iterator, std::set<std::string>::const_iterator, comparator> forward;
    std::map<std::set<std::string>::const_iterator, std::set<std::string>::const_iterator, comparator> backward;
};

void TestSimple() {
    Translator translator;
    translator.Add("fenster", "window");
    translator.Add("tisch", "table");

    assert(translator.TranslateForward("fenster") == "window");
    assert(translator.TranslateBackward("table") == "tisch");
    assert(translator.TranslateForward("table") == "");
}

int main() {
    TestSimple();
    return 0;
}

额外说明 #

• std::set的迭代器（包括const迭代器）在插入元素后不会失效，你的初始设计思路是正确的。
• 使用const_iterator符合C++的const正确性原则，彻底解决了类型不匹配的问题。
• 加入重复映射检查可以避免map存储冗余条目，进一步优化性能。

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