自定义分配器可用于控制STL容器内存管理，通过实现allocate、deallocate等接口，结合内存池提升小对象分配效率，需注意线程安全与正确性。
在C++中，STL容器（如vector、list、map等）默认使用系统提供的内存分配器来管理内存。但有时我们希望控制内存的分配方式，比如使用内存池、共享内存或自定义堆管理机制。这时就可以通过自定义allocator来实现。
理解STL分配器的基本结构
STL中的分配器是一个模板类，必须满足一定的接口要求。最简单的自定义分配器需要实现以下几个关键部分：
- value_type：容器元素类型
- allocate()：分配原始内存块
- deallocate()：释放内存
-
construct()：构造对象（可选，C++17后推荐用
std::allocator_traits） - destroy()：析构对象
一个最小可用的分配器示例如下：
template <typename T>
struct MyAllocator {
using value_type = T;
<pre class="brush:php;toolbar:false;">MyAllocator() = default;
template <typename U>
MyAllocator(const MyAllocator<U>&) {}
T* allocate(std::size_t n) {
if (n > std::numeric_limits<std::size_t>::max() / sizeof(T))
throw std::bad_array_new_length();
T* ptr = static_cast<T*>(::operator new(n * sizeof(T)));
return ptr;
}
void deallocate(T* ptr, std::size_t n) {
::operator delete(ptr);
}};
立即学习“C++免费学习笔记（深入）”；
如何将自定义分配器用于STL容器
只要分配器符合要求，就可以作为模板参数传给STL容器：
std::vector<int, MyAllocator<int>> vec; vec.push_back(10); vec.push_back(20);
注意：两个分配器类型相同的容器才能进行赋值或移动操作。若想支持不同分配器间的交换，需额外处理。
实现更高效的内存池分配器
常见需求是使用内存池减少频繁调用new/delete带来的开销。下面是一个简化版的内存池分配器框架：
template <typename T>
class PoolAllocator {
private:
struct Block {
Block* next;
};
Block* free_list = nullptr;
const size_t block_size = sizeof(T) < sizeof(Block*) ? sizeof(Block*) : sizeof(T);
<p>public:
using value_type = T;</p><pre class="brush:php;toolbar:false;">PoolAllocator() = default;
template <typename U>
PoolAllocator(const PoolAllocator<U>&) {}
T* allocate(size_t n) {
if (n != 1)
throw std::bad_alloc(); // 简化：只支持单个对象分配
if (!free_list) {
// 扩展内存池
constexpr size_t pool_size = 1024;
T* raw_mem = reinterpret_cast<T*>(::operator new(pool_size * block_size));
for (size_t i = 0; i < pool_size - 1; ++i) {
Block* b = reinterpret_cast<Block*>(raw_mem + i);
b->next = reinterpret_cast<Block*>(raw_mem + i + 1);
}
reinterpret_cast<Block*>(raw_mem + pool_size - 1)->next = nullptr;
free_list = reinterpret_cast<Block*>(raw_mem);
}
T* result = reinterpret_cast<T*>(free_list);
free_list = free_list->next;
return result;
}
void deallocate(T* p, size_t n) {
if (p) {
Block* b = reinterpret_cast<Block*>(p);
b->next = free_list;
free_list = b;
}
}};
立即学习“C++免费学习笔记（深入）”；
这个分配器预先分配一批内存块，用链表维护空闲列表，提升小对象分配效率。
注意事项与最佳实践
自定义分配器虽然灵活，但也容易出错。以下几点需要注意：
- 确保
allocate和deallocate成对工作，避免内存泄漏 - 不要在
allocate中调用构造函数，它只负责分配原始内存 - C++17起，建议依赖
std::allocator_traits提供默认的construct/destroy实现 - 多线程环境下需自行保证线程安全（比如加锁或每线程池）
- 不同类型的分配器实例之间通常不能互换，除非显式定义传播行为
基本上就这些。通过自定义分配器，你可以深度优化特定场景下的内存性能，但也要小心复杂性带来的风险。
