让自定义的容器,也能基于范围循环
C++11起,引入了基于范围的for循环这一特性,有什么好处呢?它有时可以大大地简化遍历容器的操作,比如说STL的vector。
std::vector v{1, 2, 3};
std::vector<int>::iterator it = begin(v);
for (; it != end(v); ++it)
std::cout << *it << '\n';
这是使用了迭代器的写法,这时候我们的C++11的for循环就可以大展身手了。
std::vector v{1, 2, 3};
for (const auto& item : v)
std::cout << item << '\n';
当然,除了STL,数组的遍历也是没问题的。接下来到了本篇的主线,假如说我定义了一个数据结构,如下:
template <typename _Tp, std::size_t N>
struct Container {
using value_type = _Tp;
_Tp arr[N];
std::size_t size() {
return N;
}
Container() {
for (std::size_t i{}; i < N; ++i)
arr[i] = i;
}
};
这时候当我们也想使用之前的操作的时候,我们会发现
信息有点多,不过很容易就注意到了两个关键点,分别是begin和end,我们可以为其提供一个接口看看。
template <typename _Tp, std::size_t N>
struct Container {
using value_type = _Tp;
_Tp arr[N + 1];
std::size_t size() {
return N;
}
Container() {
for (std::size_t i{}; i < N; ++i)
arr[i] = i;
}
constexpr _Tp *begin() noexcept {
return &arr[0];
}
constexpr _Tp *end() noexcept {
return &arr[N];
}
};
因为STL的end指针是指向最后一个元素的下一位,所以我们数组额外开多一个空间,经过此次修改,就能成功运行了。