使用boost库可以使用zip迭代器将已知数量的迭代器压缩在一起,但是什么时候直到运行时才能知道要压缩的迭代器数量?
为了扩展一点,我有一个大小相同的列表列表,我需要将每个索引的所有值组合在一起并将它们提供给另一个操作.现在这都是手动的,我觉得应该有更好的方法.
示例:
说我有3个列表:
我需要将这些列表转换为:
我不知道在运行时输入中有多少列表.
好吧,在花了将近 1/2 小时之后,我想出了这个dynamic_zip_iterator
可以进一步改进的类,使其看起来像 STL 式的迭代器。到目前为止,它非常具体,因为我已经std::list
在其中进行了硬编码,您可以将其替换为std::vector
或者可以使其更加通用:
无论如何,请看一下:
template<typename T>
struct dynamic_zip_iterator
{
typedef typename std::list<T>::iterator list_iterator;
std::list<list_iterator> iterators;
std::list<std::list<T>> * plists;
dynamic_zip_iterator(std::list<std::list<T>> & lists, bool isbegin) : plists(&lists)
{
auto it = plists->begin();
for( ; it != plists->end(); ++it)
{
if ( isbegin )
iterators.push_back(it->begin());
else
iterators.push_back(it->end());
}
}
dynamic_zip_iterator(const dynamic_zip_iterator & zip) :
plists(zip.plists),iterators(zip.iterators) {}
dynamic_zip_iterator operator++()
{
auto it = iterators.begin();
for( ; it != iterators.end(); ++it)
++(*it);
return *this;
}
std::list<T> operator*()
{
std::list<T> lst;
auto it = iterators.begin();
for( ; it != iterators.end(); ++it)
lst.push_back(*(*it));
return lst;
}
bool operator!=(dynamic_zip_iterator &zip)
{
auto it1 = iterators.begin();
auto it2 = zip.iterators.begin();
return (*it1) != (*it2);
}
static dynamic_zip_iterator begin(std::list<std::list<T>> & lists)
{
return dynamic_zip_iterator<T>(lists, true);
}
static dynamic_zip_iterator end(std::list<std::list<T>> & lists)
{
return dynamic_zip_iterator<T>(lists, false);
}
};
Run Code Online (Sandbox Code Playgroud)
使用它你的问题可以简化为这个函数:
std::list<std::list<int>> create_lists(std::list<std::list<int>>& lists)
{
std::list<std::list<int>> results;
auto begin = dynamic_zip_iterator<int>::begin(lists);
auto end = dynamic_zip_iterator<int>::end(lists);
for( ; begin != end ; ++begin)
{
results.push_back(*begin);
}
return results;
}
Run Code Online (Sandbox Code Playgroud)
测试代码:
int main() {
int a[] = {1, 2, 3, 4, 5}, b[] = {11, 12, 13, 14, 15}, c[] = {21, 22, 23, 24, 25};
std::list<int> l1(a,a+5), l2(b,b+5), l3(c,c+5);
std::list<std::list<int>> lists;
lists.push_back(l1);
lists.push_back(l2);
lists.push_back(l3);
std::list<std::list<int>> newlists = create_lists(lists);
for(auto lst = newlists.begin(); lst != newlists.end(); ++lst)
{
std::cout << "[";
std::copy(lst->begin(), lst->end(), std::ostream_iterator<int>(std::cout, " "));
std::cout << "]" << std::endl;
}
return 0;
}
Run Code Online (Sandbox Code Playgroud)
输出:
[1 11 21 ]
[2 12 22 ]
[3 13 23 ]
[4 14 24 ]
[5 15 25 ]
Run Code Online (Sandbox Code Playgroud)
在线演示: http: //ideone.com/3FJu1
归档时间: |
|
查看次数: |
2778 次 |
最近记录: |