C++ pmr 容器使用多态分配器示例

在现代 C++ 编程中，std::pmr（Pointer Management Resources）库提供了一种灵活且高效的方式来管理内存分配。通过使用 pmr 容器和多态分配器，可以更好地控制内存的分配和释放，从而提高程序的性能和资源利用率。

什么是 pmr 容器？

pmr 是 Pointer Management Resources 的缩写，它提供了一组基于内存资源（memory resources）的容器和分配器。这些容器和分配器允许你自定义内存分配策略，从而实现更高效的内存管理。

多态分配器的作用

多态分配器是指可以动态改变其行为的分配器。通过使用多态分配器，你可以根据不同的需求选择不同的内存分配策略，例如使用堆内存、共享内存、线程局部存储等。

使用 pmr 容器和多态分配器的步骤

以下是一个简单的示例，展示如何使用 pmr 容器和多态分配器来管理内存分配。

步骤 1: 包含必要的头文件

#include <iostream>
#include <vector>
#include <memory_resource>

步骤 2: 创建自定义内存资源

你可以创建自己的内存资源类，继承自 std::pmr::memory_resource，并重写其 do_allocate 和 do_deallocate 方法。

class CustomMemoryResource : public std::pmr::memory_resource {
public:
    void* do_allocate(std::size_t bytes, std::size_t alignment) override {
        // 自定义内存分配逻辑
        return ::operator new(bytes);
    }

    void do_deallocate(void* p, std::size_t bytes, std::size_t alignment) override {
        // 自定义内存释放逻辑
        ::operator delete(p);
    }
};

步骤 3: 创建 pmr 容器并使用自定义内存资源

你可以使用 pmr::vector 或其他 pmr 容器，并指定自定义内存资源作为其分配器。

int main() {
    CustomMemoryResource customResource;

    std::pmr::vector<int> vec(&customResource);

    for (int i = 0; i < 10; ++i) {
        vec.push_back(i);
    }

    for (const auto& num : vec) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

解释

1. 包含头文件：我们包含了必要的头文件，包括 <iostream> 用于输入输出，<vector> 用于使用 pmr::vector，以及 <memory_resource> 用于使用 pmr 相关功能。
2. 创建自定义内存资源：我们定义了一个名为 CustomMemoryResource 的类，继承自 std::pmr::memory_resource。在这个类中，我们重写了 do_allocate 和 do_deallocate 方法，实现了自定义的内存分配和释放逻辑。
3. 创建 pmr 容器并使用自定义内存资源：我们在 main 函数中创建了一个 CustomMemoryResource 对象，并将其传递给 pmr::vector 的构造函数。这样，pmr::vector 就会使用自定义内存资源来管理内存分配。

通过这种方式，你可以根据需要自定义内存分配策略，从而优化程序的内存使用效率。

总结

pmr 容器和多态分配器是 C++ 中一种强大的工具，可以帮助你更好地管理和优化内存分配。通过创建自定义内存资源并将其应用到 pmr 容器中，你可以实现更灵活和高效的内存管理策略。希望这个示例能帮助你理解如何在实际项目中使用 pmr 容器和多态分配器。