本文主要介绍C++编程语言的STL(Standard Template Library)中vector容器的相关知识,同时通过示例代码介绍vector的常见用法。
1 概述
vector是C++编程语言的STL中的内容,是同一种类型的对象的集合,每个对象都有一个对应的整数索引值,和(C++编程语言中的)string对象一样,标准库负责管理与存储元素相关的内存。
有时候会把vector称为容器,是因为它可以包含其他对象,能够存放任意类型的动态数组、增加和压缩数据。一个容器中的所有对象都必须是同一种类型的。
vector不是一种数据类型,而是一个类模板(class template)。使用模板可以编写一个类定义或函数定义,用于多种不同的数据类型。因此,可以定义保存string对象的vector、或保存int类型的vector、又或是保存自定义的类型对象的vector。
vector和built-in数组类似,拥有一段连续的内存空间,能够很好地支持即时存取(即“[]”操作符)。但是由于它的内存空间是连续的,所以在vector中间进行插入和删除会造成内存块的拷贝。另外,当插入的元素较多时,vector预留的内存空间可能不足,就需要重新申请一块足够大的内存,并把原来的数据拷贝到新的内存空间中,这些过程会影响vector的效率。虽然如此,实际上目前用的最多的还是vector容器。在大多数情况下,vector的效率还是不错的。
2 常见用法
2.1 构造vector
可以通过下面的方式构造vector:
vector<int> vecint;
也可以在构造vector的同时指定其长度并进行初始化,如下:
vector<int> vecint(3, 6);
在上面的语句中,vecInt被初始化为包含3个int类型元素、且元素值为6的vector。
创建一个“m*n”、且元素初始值为“1”的二维数组,代码如下:
vector<vector<int>> dp(m, vector<int>(n, 1));
2.2 添加元素
vector添加数据的缺省方法是push_back()。
push_back()函数会将数据添加到vector的尾部,并按需要来分配内存。用法示例如下:
vecint.push_back(i+1);
2.3 遍历元素
2.3.1 利用迭代器iterator
可以通过使用vector的迭代器iterator、调用vector对象的begin()和end()函数,实现对于vector中数据的遍历。用法示例如下:
typedef vector<int> vecInt;
vecInt vecint;
vecInt::iterator it;
for (it = vecint.begin(); it != vecint.end(); it++)
{
cout << *it << endl;
}
2.3.2 利用auto关键字
可以通过auto关键字(C++11及后续版本)实现对于vector中数据的遍历。用法示例如下:
for (auto i:vecint)
{
cout << i << endl;
}
2.4 查询元素
通常使用at方法查询(访问)元素。at方法会检查下标是否越界:如果越界,则抛出“out of range”异常。用法示例如下:
vecint.at(1)
2.5 获取元素个数
使用size()函数获取容器中的元素个数。用法示例如下:
vecint.size();
注意:size()函数的返回值类型为size_type,该类型为unsigned。因此,当涉及到size()函数与int等类型的计算时,需要特别注意unsigned与signed转换带来的数据溢出问题。例如,有下列代码:
vector<int> v;
// size() return unsigned type
int size = v.size() - 3;
cout << "[first]size of v is: " << v.size() - 3 << endl;
cout << "[second]size of v is: " << size << endl;
上方代码的输出结果如下:
通过上述结果可知,默认情况(没有强制类型转换)下,vector的size()函数参与运算时,返回的结果也是unsigned类型的。为了获取我们想要的结果,建议先调用size()函数并进行强制类型转换,再返回结果。
2.6 删除元素
使用erase(pos)方法来删除元素。erase()方法删除pos位置的数据,返回下一个数据的位置。用法示例如下:
// 删除最后一个元素
it = vecint.end();
it = it - 1; // 需要注意,(it - 1)对应最后一个元素
vecint.erase(it);
2.7 插入元素
使用insert(pos, elem)方法,在pos位置插入一个elem拷贝,返回新数据位置。用法示例如下:
// 在最后一个元素之前插入数据
it = vecint.end();
it = it - 1;
vecint.insert(it, 100);
注意:在删除和插入元素时,需要特别注意pos(即迭代器对象it)对应的元素位置。vecint.end()方法会指向最后一个元素的下一个位置,所以删除最后一个元素/在最后一个元素之前插入元素时,需要“it - 1”操作。
2.8 二维数组
可以通过下面的方式使用vector定义二维数组:
vector<vector<int>> vecvec = {{1, 2, 8}, {2, 4, 9}, {4, 7, 10}, {6, 8, 11}};
上面的代码定义了一个int类型的“3*3”的二维数组。
3 示例程序
3.1 用法示例
此处给出一个vector示例代码。代码如下:
#include <vector>
#include <iostream>
using namespace std;
int main()
{
typedef vector<int> vecInt;
vecInt vecint;
// 添加数据
for (int i = 0; i < 5; i++)
{
vecint.push_back(i+1);
}
// 遍历元素
vecInt::iterator it;
cout << "vecint is: " << endl;
for (it = vecint.begin(); it != vecint.end(); it++)
{
cout << *it << endl;
}
// 查询数据
cout << "the second element is: " << vecint.at(1) << endl;
// 删除数据,删除最后一个元素
it = vecint.end();
it = it - 1; // 需要注意,(it - 1)对应最后一个元素
vecint.erase(it);
// 再次遍历元素,观察删除操作是否成功
cout << "after del, vecint is: " << endl;
for (it = vecint.begin(); it != vecint.end(); it++)
{
cout << *it << endl;
}
// 插入数据,在最后一个元素之前插入数据
it = vecint.end();
it = it - 1;
vecint.insert(it, 100);
// 再次遍历元素,观察插入操作是否成功
cout << "after insert, vecint is: " << endl;
for (it = vecint.begin(); it != vecint.end(); it++)
{
cout << *it << endl;
}
return 0;
}
上述代码的运行结果如下:
3.2 二维数组中的查找
此处给出一个通过vector实现在二维数组中查找数字的示例代码。代码如下:
// 二维数组中的查找
// 题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按
// 照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个
// 整数,判断数组中是否含有该整数。
#include <string>
#include <iostream>
#include <vector>
bool Find(const vector<vector<int>> & vecvec, const int & rows, const int & columns, const int & number)
{
bool found = false;
if (vecvec.empty()) {
cout << "The vector is empty!" << endl;
return found;
}
if (rows > 0 && columns > 0)
{
int row = 0;
int column = columns - 1;
while (row < rows && column >= 0)
{
//cout << "row is: " << row << ", column is: " << column << endl;
//cout << "vecvec[row][column] is: " << vecvec[row][column] << endl;
if (vecvec[row][column] == number)
{
found = true;
break;
}
else if (vecvec[row][column] > number)
-- column;
else
++ row;
}
}
return found;
}
// ====================测试代码====================
void Test(const string & str, const vector<vector<int>> & vecvec, const int & rows, const int & columns, const int & number, const bool & expected)
{
cout << str << " begins:" << endl;
bool result = Find(vecvec, rows, columns, number);
if (result == expected)
{
cout << "Passed." << endl;
}
else
{
cout << "Failed." << endl;
}
}
// 1 2 8 9
// 2 4 9 12
// 4 7 10 13
// 6 8 11 15
// 要查找的数在数组中
void Test1()
{
const vector<vector<int>> & vecvec = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
Test("Test1", vecvec, 4, 4, 7, true);
}
// 要查找的数不在数组中
void Test2()
{
const vector<vector<int>> & vecvec = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
Test("Test2", vecvec, 4, 4, 5, false);
}
// 要查找的数是数组中最小的数字
void Test3()
{
const vector<vector<int>> & vecvec = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
Test("Test3", vecvec, 4, 4, 1, true);
}
// 要查找的数是数组中最大的数字
void Test4()
{
const vector<vector<int>> & vecvec = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
Test("Test4", vecvec, 4, 4, 15, true);
}
// 要查找的数比数组中最小的数字还小
void Test5()
{
const vector<vector<int>> & vecvec = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
Test("Test5", vecvec, 4, 4, 0, false);
}
// 要查找的数比数组中最大的数字还大
void Test6()
{
const vector<vector<int>> & vecvec = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
Test("Test6", vecvec, 4, 4, 16, false);
}
// 鲁棒性测试,输入空指针
void Test7()
{
const vector<vector<int>> & vecvec = {};
Test("Test7", vecvec, 0, 0, 16, false);
}
int main()
{
Test1();
Test2();
Test3();
Test4();
Test5();
Test6();
Test7();
return 0;
}
上述代码的运行结果如下: