Pythonのenumerate
のように、C++ 11 foreachループで現在のコンテナーエントリのインデックスを取得する便利な方法はありますか。
for idx, obj in enumerate(container):
pass
インデックスなども返すことができるイテレータを想像できます。
もちろん、カウンターを持つこともできますが、多くの場合、イテレーターはコンテナーに対して反復する順序を保証しません。
リクエストされた機能の適切な実装は、次の場所にあります。
https://github.com/ignatz/Pythonic
背後にある考え方は、カウントを行うカスタムイテレータを使用してラッパー構造体を作成することです。以下は、アイデアを説明するための非常に最小限の例示的な実装です。
#include <iostream>
#include <vector>
#include <Tuple>
// Wrapper class
template <typename T>
class enumerate_impl
{
public:
// The return value of the operator* of the iterator, this
// is what you will get inside of the for loop
struct item
{
size_t index;
typename T::value_type & item;
};
typedef item value_type;
// Custom iterator with minimal interface
struct iterator
{
iterator(typename T::iterator _it, size_t counter=0) :
it(_it), counter(counter)
{}
iterator operator++()
{
return iterator(++it, ++counter);
}
bool operator!=(iterator other)
{
return it != other.it;
}
typename T::iterator::value_type item()
{
return *it;
}
value_type operator*()
{
return value_type{counter, *it};
}
size_t index()
{
return counter;
}
private:
typename T::iterator it;
size_t counter;
};
enumerate_impl(T & t) : container(t) {}
iterator begin()
{
return iterator(container.begin());
}
iterator end()
{
return iterator(container.end());
}
private:
T & container;
};
// A templated free function allows you to create the wrapper class
// conveniently
template <typename T>
enumerate_impl<T> enumerate(T & t)
{
return enumerate_impl<T>(t);
}
int main()
{
std::vector<int> data = {523, 1, 3};
for (auto x : enumerate(data))
{
std::cout << x.index << ": " << x.item << std::endl;
}
}
次のような簡単な解決策はどうですか?
int counter=0;
for (auto &val: container)
{
makeStuff(val, counter);
counter++;
}
スコープを追加することで、カウンターの後にコードを追加するのが少し難しくなります。
int counter=0;
for (auto &val: container)
{{
makeStuff(val, counter);
}counter++;}
@ graham.reedsが指摘したように、通常のfor
ループも解決策です。
int counter=0;
for (auto it=container.begin(); it!=container.end(); ++it, ++counter)
{
makeStuff(val, counter);
}
そして最後に、アルゴリズムを使用した別の方法:
int counter = 0;
std::for_each(container.begin(), container.end(), [&counter](int &val){
makeStuff(val, counter++);
});
注:範囲ループと通常ループの間の順序は、標準6.5.4によって保証されています。カウンターがコンテナー内の位置と一致することを意味します。
Boostにアクセスできる場合は、その範囲アダプターを次のように使用できます。
using namespace boost::adaptors;
for (auto const& elem : container | indexed(0))
{
std::cout << elem.index() << " - " << elem.value() << '\n';
}
ソース (他の例もあります)
C++ 17と構造化バインディングはこれで問題ありません。ローカルの[i = 0](Element&) mutable
や、おそらくすべてがfor_each()
に限定されるわけではないことを認める前に行った醜い可変ラムダよりも確かに優れています。 et al。-for
ループ外のスコープを持つカウンターを必要とする他のソリューションよりも。
for (auto [it, end, i] = std::Tuple{container.cbegin(), container.cend(), 0};
it != end; ++it, ++i)
{
// something that needs both `it` and `i`ndex
}
このパターンを十分に頻繁に使用する場合、これを汎用的にすることができます。
template <typename Container>
auto
its_and_idx(Container&& container)
{
return std::Tuple{begin(container), end(container), 0};
}
// ...
for (auto [it, end, i] = its_and_idx(foo); it != end; ++it, ++i)
{
// something
}
インデックスが必要な場合は、従来のforが完全に機能します。
for (int idx=0; idx<num; ++idx)
{
// do stuff
}