自定义内存分配器可提升性能并减少内存碎片,通过实现包含value_type、allocate、deallocate等成员的模板类,遵循STL接口规范,结合rebind机制适配不同类型,可用于vector、map等容器,如使用malloc/free或对象池优化内存管理。

在c++中,为STL容器提供自定义内存分配器(Allocator)可以提升性能、减少内存碎片,或满足特定内存管理需求。标准库中的容器如 vector、list、map 等都支持通过模板参数传入自定义的 Allocator。实现一个自定义 Allocator 并不复杂,但需要遵循一定的接口规范。
理解 Allocator 的基本结构
自定义 Allocator 需要满足 STL 的接口要求。一个最简化的 Allocator 模板类通常包含以下关键成员:
- value_type:容器元素类型
- pointer:指向元素的指针类型
- const_pointer:常量指针类型
- reference:引用类型
- const_reference:常量引用类型
- size_type:大小类型(通常是 std::size_t)
- difference_type:指针差值类型
- allocate(n):分配 n 个对象的原始内存
- deallocate(p, n):释放从 p 开始的 n 个对象内存
- construct(p, args…):在 p 指向的位置构造对象
- destroy(p):析构 p 指向的对象
- rebind:允许 Allocator 适配不同类型的容器
从 C++17 起,construct 和 destroy 可以省略,由 std::allocator_traits 提供默认实现,但我们仍建议了解其作用。
实现一个简单的自定义 Allocator
下面是一个基于 malloc 和 free 的简单自定义 Allocator 示例,适用于任意类型 T:
立即学习“C++免费学习笔记(深入)”;
template <typename T> class SimpleAllocator { public: using value_type = T; using pointer = T*; using const_pointer = const T*; using reference = T&; using const_reference = const T&; using size_type = std::size_t; using difference_type = std::ptrdiff_t; <pre class='brush:php;toolbar:false;'>template <typename U> struct rebind { using other = SimpleAllocator<U>; }; SimpleAllocator() = default; template <typename U> SimpleAllocator(const SimpleAllocator<U>&) {} pointer allocate(size_type n) { if (n == 0) return nullptr; void* ptr = std::malloc(n * sizeof(T)); if (!ptr) throw std::bad_alloc(); return static_cast<pointer>(ptr); } void deallocate(pointer p, size_type) { if (p) std::free(p); } template <typename U, typename... Args> void construct(U* p, Args&&... args) { new(p) U(std::forward<Args>(args)...); } template <typename U> void destroy(U* p) { p->~U(); }
};
这个分配器使用 C 风格的 malloc/free 进行内存管理,适用于大多数基础场景。注意 rebind 的实现,它让 STL 在内部类型变化时能正确生成新的 Allocator 类型。
将自定义 Allocator 应用于 STL 容器
使用自定义 Allocator 很简单,只需将其作为模板参数传入容器即可:
std::vector<int, SimpleAllocator<int>> vec; vec.push_back(10); vec.push_back(20); <p>// 或者定义别名简化写法 using MyVector = std::vector<int, SimpleAllocator<int>>; MyVector myVec;</p>
所有标准容器都支持这种语法。例如,也可以用于 map:
std::map<int, std::string, std::less<int>, SimpleAllocator<std::pair<const int, std::string>>> myMap;
高级用法:对象池或栈内存分配
更高效的自定义 Allocator 可以实现内存池、固定块分配或线程局部存储。例如,预先分配一大块内存,在 allocate 时从中切分,避免频繁调用系统调用。
这类优化适合频繁创建销毁小对象的场景,比如游戏开发或高频交易系统。你可以维护一个空闲链表,在 allocate 时返回空闲块,deallocate 时不真正释放,而是放回池中。
需要注意的是,自定义 Allocator 必须保证线程安全,如果多线程同时访问同一容器。此外,C++17 后推荐使用 std::pmr::memory_resource 和 std::pmr::polymorphic_allocator 来实现更灵活的内存管理策略,它们提供了更高层次的抽象。
基本上就这些。实现一个可用的自定义 Allocator 不难,关键是理解 STL 的内存管理机制和类型绑定方式。根据实际需求选择简单封装或高性能池化方案,能有效提升程序效率。