C++ std::unique_ptr数组版本的operator[]访问

在C++中，std::unique_ptr 是一种管理动态分配对象所有权的智能指针。它确保在指针生命周期结束时，所指向的对象会被自动销毁，从而有效防止内存泄漏。除了管理单个对象，std::unique_ptr 也可以管理动态数组，即 std::unique_ptr<T[]>。本文将详细介绍如何使用 std::unique_ptr<T[]> 的 operator[] 来访问数组元素。

1. 声明和初始化 std::unique_ptr 数组

要使用 std::unique_ptr 管理数组，你需要包含 <memory> 头文件。声明和初始化的方式有两种，推荐使用 std::make_unique，因为它更安全且代码更简洁。

1.1 使用 new 关键字（不推荐）

你可以直接使用 new 关键字分配数组，并将其传递给 std::unique_ptr 的构造函数。

#include <memory>

// 声明一个指向包含10个int类型元素的数组的智能指针
std::unique_ptr<int[]> my_array(new int[10]);

注意：这种方式在C++11中可用，但从C++14开始，std::make_unique 被引入并成为更优的选择。

1.2 使用 std::make_unique（推荐）

std::make_unique 是C++14引入的工厂函数，它能更安全地创建智能指针，并自动处理异常情况，避免内存泄漏。

#include <memory>

// 使用auto类型推导，创建一个包含10个int类型元素的数组
auto my_array = std::make_unique<int[]>(10);

2. 使用 operator[] 访问数组元素

std::unique_ptr<T[]> 重载了 operator[]，使得你可以像使用普通C风格数组一样访问其元素。operator[] 返回一个对数组元素的引用，因此你可以对其进行读写操作。

2.1 读取元素

要读取数组中特定索引的值，直接使用 operator[] 即可。

// 假设 my_array 已经通过 std::make_unique<int[]>(10) 初始化
int first_element = my_array[0]; // 读取索引为0的元素
int fifth_element = my_array[4]; // 读取索引为4的元素

2.2 写入元素

要修改数组中特定索引的值，同样使用 operator[]，然后进行赋值。

// 假设 my_array 已经通过 std::make_unique<int[]>(10) 初始化
my_array[0] = 100; // 将索引为0的元素设置为100
my_array[9] = 999; // 将索引为9的元素设置为999

2.3 遍历数组

你可以使用 for 循环结合 operator[] 来遍历整个数组。

// 假设 my_array 已经通过 std::make_unique<int[]>(10) 初始化
for (int i = 0; i < 10; ++i) {
    my_array[i] = i * i; // 为每个元素赋值
}

3. 重要注意事项：边界检查

与标准库中的 std::vector 不同，std::unique_ptr<T[]> 的 operator[] 不会进行边界检查。这意味着如果你访问一个超出数组范围的索引，程序的行为是未定义的，可能导致程序崩溃或数据损坏。

// 假设 my_array 是一个大小为10的数组
int value = my_array[10]; // 错误！索引10超出了数组范围（有效索引是0-9）
my_array[20] = 5;        // 错误！同样会导致未定义行为

最佳实践：始终确保你使用的索引在数组的有效范围内（即 0 到 size - 1）。如果你需要自动的边界检查，应考虑使用 std::vector。

4. 内存自动管理

std::unique_ptr 的最大优势在于其自动内存管理。当你声明的 std::unique_ptr<T[]> 变量离开其作用域时，其析构函数会被自动调用，从而释放它所管理的数组内存。你无需手动调用 delete[]。

#include <iostream>
#include <memory>

void create_and_use_array() {
    // 在函数内部创建一个智能指针
    auto my_array = std::make_unique<int[]>(5);

    // 使用数组
    for (int i = 0; i < 5; ++i) {
        my_array[i] = i + 1;
    }

    // 函数结束时，my_array 离开作用域，数组内存被自动释放
    // 无需编写 delete[] my_array;
}

int main() {
    create_and_use_array(); // 调用函数
    // 函数返回后，数组内存已安全释放
    return 0;
}

5. std::unique_ptr 数组 vs. std::vector

虽然两者都能管理动态数组，但它们在功能和使用场景上有所不同。下表对比了它们的主要特性。

选择建议：

• 如果你需要一个固定大小的数组，并且希望利用智能指针的自动内存管理，std::unique_ptr<T[]> 是一个很好的选择。
• 如果你需要一个可以动态增长或收缩的数组，并且希望有内置的边界检查，std::vector 是更安全、更灵活的选择。

6. 常见错误

1. 使用 . 操作符：std::unique_ptr<T[]> 是一个指针，应使用 -> 或 []，而不是 .。

   // 错误
   my_array.0 = 10; 
   // 正确
   my_array[0] = 10;

2. 忘记指定数组大小：使用 new 时必须指定大小。

   // 错误
   std::unique_ptr<int[]> ptr(new int[]); 
   // 正确
   std::unique_ptr<int[]> ptr(new int[10]);

3. 混淆 std::unique_ptr<T> 和 std::unique_ptr<T[]>：std::unique_ptr<T> 用于单个对象，其析构函数调用 delete；std::unique_ptr<T[]> 用于数组，其析构函数调用 delete[]。不要混用。