it编程 > 编程语言 > C/C++

C++ std::Set<std::pair>的实现示例

10人参与 2025-10-21 C/C++

1)std::set的三个关键特性

2) 关于比较:timestamp与std::unique_ptr<int>能否比较?

字典序:std::pair<a,b> 的默认 < 比较是先比较 first,若相等再比较 second。因此在 std::set<std::pair<timestamp, std::unique_ptr<int>>> 中:

  1. 先比较 timestamp;
  2. timestamp 相等时,再比较 unique_ptr<int> 的地址。

3) 自定义透明比较器(heterogeneous lookup)

直接用 pair<timestamp, unique_ptr<int>> 做键会遇到查找难题:

透明比较器(具备 using is_transparent = void;)允许 set.find / erase / lower_bound 等直接用可比较但类型不同的键(如 const int*)进行 o(log n) 查找。

4) crud 规范做法(c++17+)

下述代码块分别演示 create/read/update/delete。请先看类型与比较器。

#include <set>
#include <memory>
#include <utility>
#include <optional>
#include <iostream>

// ========== 1) 可排序的 timestamp ==========
class timestamp {
    int secondsfromepoch_; // 私有
public:
    explicit timestamp(int s = 0) : secondsfromepoch_(s) {}
    int seconds() const noexcept { return secondsfromepoch_; }
    // 定义严格弱序(仅按秒比较)
    friend bool operator<(const timestamp& a, const timestamp& b) noexcept {
        return a.secondsfromepoch_ < b.secondsfromepoch_;
    }
};

// 方便书写
using key = std::pair<timestamp, std::unique_ptr<int>>;

// ========== 2) 透明比较器:支持 pair<timestamp, unique_ptr<int>>
// 以及 (timestamp, const int*) 这种异质键的比较与查找 ==========
struct paircmp {
    using is_transparent = void; // 启用异质查找

    // 基本:pair vs pair(按字典序:先时间戳,再指针地址)
    bool operator()(const key& a, const key& b) const noexcept {
        if (a.first < b.first) return true;
        if (b.first < a.first) return false;
        return std::less<const int*>{}(a.second.get(), b.second.get());
    }

    // 异质:pair vs (timestamp, const int*)
    bool operator()(const key& a, const std::pair<timestamp, const int*>& b) const noexcept {
        if (a.first < b.first) return true;
        if (b.first < a.first) return false;
        return std::less<const int*>{}(a.second.get(), b.second);
    }
    bool operator()(const std::pair<timestamp, const int*>& a, const key& b) const noexcept {
        if (a.first < b.first) return true;
        if (b.first < a.first) return false;
        return std::less<const int*>{}(a.second, b.second.get());
    }

    // 也可追加只按 timestamp 的异质比较(用于范围查询)
    bool operator()(const key& a, const timestamp& t) const noexcept {
        return a.first < t;
    }
    bool operator()(const timestamp& t, const key& b) const noexcept {
        return t < b.first;
    }
};

using pairset = std::set<key, paircmp>;

a. create(插入)

pairset s;

// 1) 直接 emplace(推荐)
s.emplace(timestamp{100}, std::make_unique<int>(42));
s.emplace(timestamp{100}, std::make_unique<int>(7));   // 与上一个可共存:时间戳相同但地址不同
s.emplace(timestamp{120}, std::make_unique<int>(99));

// 2) 先构造 key 再 move
key k{ timestamp{130}, std::make_unique<int>(123) };
s.insert(std::move(k)); // k.second 置空

要点

b. read(查询/遍历)

1) 按(timestamp, 指针地址)精确查找

利用透明比较器,避免构造临时 unique_ptr

timestamp t{100};
const int* addr = /* 已知的底层指针地址,如 it->second.get() */;
auto it = s.find(std::pair<timestamp, const int*>{t, addr});
if (it != s.end()) {
    std::cout << "found value=" << *(it->second) << "\n";
}

2) 按时间戳做范围/等值查找

// 所有 timestamp == 100 的区间:
auto rng = s.equal_range(timestamp{100}); // 依赖我们在比较器中提供的 (key, timestamp) 重载
for (auto it = rng.first; it != rng.second; ++it) {
    // 这些元素的 it->first.seconds() 都是 100
}

// 所有 timestamp 在 [100, 130):
for (auto it = s.lower_bound(timestamp{100}); it != s.lower_bound(timestamp{130}); ++it) {
    // ...
}

3) 遍历(有序)

for (const auto& [ts, ptr] : s) {
    std::cout << ts.seconds() << " -> " << (ptr ? *ptr : -1) << "\n";
}

c. update(更新)

重要原则std::set 中的元素作为“键”不可就地修改其影响排序的部分(包括 timestampunique_ptr 的地址)。否则会破坏红黑树不变量。
正确做法:使用 node handle(c++17)进行“摘除-修改-再插入”。

1) 修改timestamp或替换unique_ptr(地址会变)

// 找到一个元素
auto it = s.lower_bound(timestamp{100});
if (it != s.end() && it->first.seconds() == 100) {
    // 1) extract 节点,容器不再管理平衡关系中的该节点
    auto nh = s.extract(it);       // node_handle,拥有该 pair 的完整所有权
    // 2) 修改 key 内容(注意:任何影响排序的字段都只能在 node 中修改)
    nh.value().first = timestamp{105};                  // 改时间戳
    nh.value().second = std::make_unique<int>(555);     // 新指针(地址变化)
    // 3) 重新插入
    auto [pos, ok] = s.insert(std::move(nh));
    // ok==false 表示与现有元素等价(违反唯一性),插入失败
}

2) 仅更新指向对象的“值”(不改变地址)

如果你不更换 unique_ptr 本身,只是修改它指向的 int 的数值(地址不变),就不会影响排序,可在常量迭代器上做“逻辑修改”前需要去除 const:

简化起见,建议:提取 node 后修改再插回,语义最清晰。

d. delete(删除)

// 1) 迭代器删除
if (!s.empty()) {
    s.erase(s.begin());
}

// 2) 按异质键删除(timestamp + 地址)
timestamp t{100};
const int* addr = /*...*/;
s.erase(std::pair<timestamp, const int*>{t, addr});

// 3) 按时间戳范围删除
s.erase(s.lower_bound(timestamp{100}), s.lower_bound(timestamp{130}));

5) 典型陷阱与建议

  1. 临时 unique_ptr 作为查找键:千万不要用 find({ts, std::unique_ptr<int>(raw)}) 查找,临时 unique_ptr 析构时会 delete raw,导致双重释放。请使用透明比较器 + 原始指针地址的异质查找。

  2. 修改键值破坏有序性:在容器中直接改 timestamp 或把 unique_ptr 换成另一块地址,都会破坏树的排序假设。务必用 extract→修改→insert

  3. 语义核对unique_ptr 的比较是按地址而非按“指向内容”。如果你想让“同一 timestamp + 相同内容(例如 *ptr)才算相等,需要自定义比较器改成按 *ptr 值比较(并处理空指针)。

  4. 标准版本

    • c++17 起:有 node_handle、更明确的对仅移动键(如 unique_ptr)的支持。强烈建议使用 c++17+。
    • 透明比较器用法在 c++14 就可行(is_transparent 习惯用法),但与 node handle 结合最顺畅的是 c++17+。

6) 小结(要点清单)

完整最小示例(可直接参考)

#include <set>
#include <memory>
#include <utility>
#include <iostream>

class timestamp {
    int secondsfromepoch_;
public:
    explicit timestamp(int s = 0) : secondsfromepoch_(s) {}
    int seconds() const noexcept { return secondsfromepoch_; }
    friend bool operator<(const timestamp& a, const timestamp& b) noexcept {
        return a.secondsfromepoch_ < b.secondsfromepoch_;
    }
};

using key = std::pair<timestamp, std::unique_ptr<int>>;

struct paircmp {
    using is_transparent = void;

    bool operator()(const key& a, const key& b) const noexcept {
        if (a.first < b.first) return true;
        if (b.first < a.first) return false;
        return std::less<const int*>{}(a.second.get(), b.second.get());
    }
    bool operator()(const key& a, const std::pair<timestamp, const int*>& b) const noexcept {
        if (a.first < b.first) return true;
        if (b.first < a.first) return false;
        return std::less<const int*>{}(a.second.get(), b.second);
    }
    bool operator()(const std::pair<timestamp, const int*>& a, const key& b) const noexcept {
        if (a.first < b.first) return true;
        if (b.first < a.first) return false;
        return std::less<const int*>{}(a.second, b.second.get());
    }
    bool operator()(const key& a, const timestamp& t) const noexcept { return a.first < t; }
    bool operator()(const timestamp& t, const key& b) const noexcept { return t < b.first; }
};

using pairset = std::set<key, paircmp>;

int main() {
    pairset s;

    // create
    auto [it1, ok1] = s.emplace(timestamp{100}, std::make_unique<int>(42));
    auto [it2, ok2] = s.emplace(timestamp{100}, std::make_unique<int>(7));
    s.emplace(timestamp{120}, std::make_unique<int>(99));

    // read: find by (timestamp, raw pointer)
    const int* addr = it1->second.get();
    auto it = s.find(std::pair<timestamp, const int*>{timestamp{100}, addr});
    if (it != s.end()) {
        std::cout << "found ts=" << it->first.seconds() << " val=" << *it->second << "\n";
    }

    // read: range by timestamp
    std::cout << "ts==100:\n";
    for (auto p = s.lower_bound(timestamp{100}); p != s.upper_bound(timestamp{100}); ++p) {
        std::cout << "  " << p->first.seconds() << " -> " << *p->second << "\n";
    }

    // update: extract -> modify -> insert
    if (it2 != s.end()) {
        auto nh = s.extract(it2);                        // 取出节点
        nh.value().first = timestamp{105};               // 改键(时间戳)
        nh.value().second = std::make_unique<int>(555);  // 换指针(地址变)
        s.insert(std::move(nh));                         // 重新插入
    }

    // delete: by heterogeneous key
    s.erase(std::pair<timestamp, const int*>{timestamp{100}, addr});

    // 遍历
    for (const auto& [ts, ptr] : s) {
        std::cout << ts.seconds() << " -> " << (ptr ? *ptr : -1) << "\n";
    }
}

到此这篇关于c++ std::set<std::pair>的实现示例的文章就介绍到这了,更多相关c++ std::set<std::pair>内容请搜索代码网以前的文章或继续浏览下面的相关文章希望大家以后多多支持代码网!

(0)

您想发表意见!!点此发布评论

推荐阅读

C++ STL中容器string超详细讲解

10-21

C语言memcpy函数用法详解:高效内存复制的实用工具

10-21

使用C++设计开发一个功能完善的多进程管理器

10-22

C/C++ Qt监控文件状态变化方式

10-22

C++之Set/Get使用及说明

10-22

C++玫瑰花字符画项目实战教程

10-22

猜你喜欢

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论