使用 C++ STL 时如何进行内存优化？

使用以下优化策略可以优化 c++ stl 中的内存使用：1. 使用自定义分配器控制内存分配方式；2. 使用 reserve() 预分配空间，避免动态内存分配开销；3. 运用搬移语义或引用语义，避免不必要的内存复制。

在 C++ STL 中进行内存优化

STL（标准模板库）是 C++ 中广泛使用的库，提供了一组高效且经过良好测试的数据结构和算法。然而，在使用 STL 时，不当的内存管理可能会导致性能问题。以下是一些优化内存使用的技巧：

使用自定义分配器

立即学习“C++免费学习笔记（深入）”；

可以通过提供自定义分配器来控制 STL 容器分配内存的方式。自定义分配器可以实现各种优化策略，例如：

// 自定义分配器用于使用内存池分配内存
class MyAllocator {
    std::vector memory_pool;
public:
    void* allocate(std::size_t size) {
        if (memory_pool.size() >= size) {
            void* ptr = &memory_pool[0];
            memory_pool.erase(memory_pool.begin());
            return ptr;
        }
        return std::malloc(size);
    }
    void deallocate(void* ptr, std::size_t size) {
        // 将内存返回到池中
        memory_pool.push_back(*static_cast(ptr));
    }
};

通过将 MyAllocator 传递给容器构造函数，我们可以使用自定义分配策略：

Bika.ai

打造您的AI智能体员工团队

下载

std::vector my_vector;

使用容器大小优化

STL 容器通常使用动态内存分配，因此预分配足够的空间至关重要。使用 reserve() 方法可以预分配给定数量的元素：

std::vector my_vector;
my_vector.reserve(100);

避免不必要的复制

STL 算法和容器操作可以创建新对象，导致不必要的内存复制。为了避免这种情况，可以使用搬移语义或引用语义。例如，使用 std::move() 将元素移动到容器中，而不是复制：

std::vector my_vector;
my_vector.push_back(std::move(my_value));

实战案例

以下示例演示了如何使用自定义分配器优化内存分配：

#include 
#include 

// 自定义分配器使用内存池分配内存
class MyAllocator : public std::allocator {
    std::vector memory_pool;
public:
    MyAllocator() {}
    MyAllocator(const MyAllocator&) = default;
    template
    MyAllocator(const MyAllocator&) {}
    int* allocate(std::size_t n) {
        if (n <= memory_pool.size()) {
            int* ptr = &memory_pool[0];
            memory_pool.erase(memory_pool.begin());
            return ptr;
        }
        return std::allocator::allocate(n);
    }
    void deallocate(int* ptr, std::size_t) {
        // 将内存返回到池中
        memory_pool.push_back(*ptr);
        std::allocator::deallocate(ptr, 1);
    }
};

int main() {
    // 使用自定义分配器创建 vector
    std::vector my_vector;

    // 分配 1000 个元素
    my_vector.reserve(1000);

    // 使用自定义分配器分配的内存的效率更高
    return 0;
}