C++ STL 中distance() 函数用法

C++ STL 中distance() 函数用法

对于作用与同一容器的两个同类型迭代器可以有效指定一个区间范围，要想获取该指定范围内包含元素的个数，就需要借助distance()函数。

template<class InputIterator> typename iterator_traits<InputIterator>::difference_type distance(InputIterator first, InputIterator last);

其中，first 和 last 都为迭代器，其类型可以是输入迭代器、前向迭代器、双向迭代器以及随机访问迭代器；该函数会返回[first, last)范围内包含的元素的个数。

first 和 last 的迭代器类型，直接决定了 distance() 函数底层的实现机制：

• 当 first 和 last 为随机访问迭代器时，distance() 底层直接采用 last - first 求得 [first, last) 范围内包含元素的个数，其时间复杂度为O(1)常数阶；
• 当 first 和 last 为非随机访问迭代器时，distance() 底层通过不断执行 ++first（或者 first++）直到 first==last，由此来获取 [first, last) 范围内包含元素的个数，其时间复杂度为O(n)线性阶。

另外，distance() 函数定义在<iterator>头文件，并位于 std 命名空间中。

示例：

#include <iostream>
#include <iterator>
#include <vector>

using namespace std;

int main() {
    vector<int> test;
    // 向空的vector容器中添加元素0~9
    for (int i = 0; i < 10; ++i) {
        test.push_back(i);
    }
    
    vector<int>::iterator first = test.begin();	// 指向元素0
    vector<int>::iterator last = test.end();	// 指向元素9之后的位置
    // 获取[first, last)范围内包含的元素个数
    cout << "distance() = " << distance(first, last) << endl;	
    // 结果为：distance() = 10
    
    return 0;
}