迹忆客 专注技术分享

本文将介绍在 C++ 中如何调整数组大小的多种方法。

使用 resize 方法在 C++ 中调整一个数组的大小

由于定长数组容器在 C++ 中是不应该调整大小的，所以我们将重点讨论 std::vector 类。resize 是 vector 容器的内置函数，它可以改变向量所包含的元素数。如果第一个参数 count 小于 vector 的当前大小，该函数可以减少元素的数量。否则，如果 count 参数大于向量大小，resize 会插入额外的元素，默认值为 0，或者用户可以提供所需的值作为函数的第二个参数。

#include <iostream>
#include <vector>

using std::cout; using std::cin;
using std::endl; using std::string;
using std::vector;

template<typename T>
void printVectorElements(vector<T> &vec)
{
    for (auto i = 0; i < vec.size(); ++i) {
        cout << vec.at(i) << "; ";
    }
    cout << endl;
}

int main() {
    vector<int> i_vec1 = {12, 32, 43, 53, 23, 65, 84};

    cout << "i_vec1             : ";
    printVectorElements(i_vec1);
    cout << "size: " << i_vec1.size() << endl;

    i_vec1.resize(i_vec1.size() - 2);

    cout << "i_vec1 (resized -2): ";
    printVectorElements(i_vec1);
    cout << "size: " << i_vec1.size() << endl;
    cout << endl;

    return EXIT_SUCCESS;
}

输出：

i_vec1             : 12; 32; 43; 53; 23; 65; 84;
size: 7
i_vec1 (resized -2): 12; 32; 43; 53; 23;
size: 5

使用 erase 方法在 C++ 中减少数组中的元素数量

erase 函数是 std::vector 类的另一个内置方法，它可以从 vector 中删除单个元素，甚至删除由相应迭代器指定的整个范围。

在下面的例子中，从元素 2 到末端的给定范围被从整数的 vector 中删除。注意，通常的做法是用 begin/end 迭代器传递范围参数。

#include <iostream>
#include <vector>

using std::cout; using std::cin;
using std::endl; using std::string;
using std::vector;

template<typename T>
void printVectorElements(vector<T> &vec)
{
    for (auto i = 0; i < vec.size(); ++i) {
        cout << vec.at(i) << "; ";
    }
    cout << endl;
}

int main() {
    vector<int> i_vec1 = {12, 32, 43, 53, 23, 65, 84};

    cout << "i_vec1          : ";
    printVectorElements(i_vec1);
    cout << "size: " << i_vec1.size() << endl;

    i_vec1.erase(i_vec1.begin() + 2, i_vec1.end());

    cout << "i_vec1 (resized): ";
    printVectorElements(i_vec1);
    cout << "size: " << i_vec1.size() << endl;
    cout << endl;

    return EXIT_SUCCESS;
}

输出：

i_vec1          : 12; 32; 43; 53; 23; 65; 84;
size: 7
i_vec1 (resized): 12; 32;
size: 2

使用自定义函数在 C++ 中调整数组的大小

另外，我们也可以定义一个单独的函数，它可以遍历向量，并从向量的末端删除给定数量的元素。这可以使用 pop_back 内置函数来实现，它可以删除 vector 中的最后一个元素。如接下来的代码示例所示，定义了 resizeVector 函数模板，用来接收一个类型为 T 的向量和要从其中移除的若干元素。

#include <iostream>
#include <vector>

using std::cout; using std::cin;
using std::endl; using std::string;
using std::vector;

template<typename T>
void printVectorElements(vector<T> &vec)
{
    for (auto i = 0; i < vec.size(); ++i) {
        cout << vec.at(i) << "; ";
    }
    cout << endl;
}

template<typename T>
void resizeVector(vector<T> &vec, int elems)
{
    for (auto i = 0; i < elems; ++i) {
        vec.pop_back();
    }
}

int main() {
    vector<int> i_vec1 = {12, 32, 43, 53, 23, 65, 84};

    cout << "i_vec1          : ";
    printVectorElements(i_vec1);
    cout << "size: " << i_vec1.size() << endl;

    resizeVector(i_vec1, 3);

    cout << "i_vec1 (resized): ";
    printVectorElements(i_vec1);
    cout << "size: " << i_vec1.size() << endl;
    cout << endl;

    return EXIT_SUCCESS;
}

输出：

i_vec1          : 12; 32; 43; 53; 23; 65; 84;
size: 7
i_vec1 (resized): 12; 32; 43; 53;
size: 4