我有一个函数,我需要一个迭代器的底层数据类型作为返回类型,如下所示:
#include <iostream>
#include <vector>
#include <iterator>
template<class T, class ForwardIterator>
std::vector<T> get_odd(ForwardIterator data_begin, ForwardIterator data_end)
{
std::vector<T> ret;
std::copy_if(data_begin, data_end, std::back_inserter(ret), [](int x) { return x % 2; });
return ret;
}
int main()
{
std::vector<int> vi { 1, 2, 3, 4, 5 };
for (auto i : get_odd<int>(vi.begin(), vi.end()))
std::cout << i << ", ";
std::cout << std::endl;
std::vector<unsigned int> vui{ 9UL, 8UL, 7UL, 6UL, 5UL };
// Note the 'char' I provided to the function, this will print weird chars
for (auto i : get_odd<char>(vui.begin(), vui.end()))
std::cout << i << ", ";
std::cout << std::endl;
return 0;
}
在main()中,我必须向get_odd函数显式提供数据类型'int'或'char',有没有办法找出迭代器的底层数据类型,让模板自动扣除正确的数据类型?
答案 0 :(得分:6)
可以使用std::iterator_traits
查询迭代器属性:
template<class ForwardIterator>
std::vector<typename std::iterator_traits<ForwardIterator>::value_type>
get_odd(ForwardIterator data_begin, ForwardIterator data_end)
{
std::vector<typename std::iterator_traits<ForwardIterator>::value_type> ret;
std::copy_if(data_begin, data_end, std::back_inserter(ret), [](int x) { return x % 2; });
return ret;
}
答案 1 :(得分:2)
如果您的编译器支持C ++ 11,您可以使用:
#include <type_traits>
// ...
typename std::remove_reference<decltype(*data_begin)>::type
为方便起见,可能typedef
:
typedef typename std::remove_reference<decltype(*data_begin)>::type ContainedType;
std::vector<ContainedType> ret;
答案 2 :(得分:2)
这经常出现,我使用方便的模板别名来包装iterator_traits
:
template <typename I>
using ValueType = typename std::iterator_traits<I>::value_type;
template<class ForwardIterator>
std::vector<ValueType<ForwardIterator>>
get_odd(ForwardIterator data_begin, ForwardIterator data_end)
{
std::vector<ValueType<ForwardIterator>> ret;
std::copy_if(data_begin, data_end, std::back_inserter(ret), [](int x) {
return x % 2;
});
return ret;
}
答案 3 :(得分:0)
template<class ForwardIterator>
std::vector<typename std::iterator_traits<ForwardIterator>::value_type> get_odd(ForwardIterator data_begin, ForwardIterator data_end)
{
....
}