如何用迭代器特性专门化函数模板

Question

template <typename ForwIt>
typename std::iterator_traits<ForwIt>::value_type
foo(ForwIt begin, ForwIt end, double bar)
{
using value_type = typename std::iterator_traits<ForwIt>::value_type;
// value_type is the type of the values the iterator ForIt points to
value_type baz;
// Do stuff with the values in range [begin, end).
// And modify baz;

return baz;
}
int main()
{
std::vector<int> numbers;
for (int i = 0; i < 10; ++i)
numbers.push_back(i);

const std::vector<int> v0(numbers.begin(), numbers.end());
const std::vector<float> v1(numbers.begin(), numbers.end());
std::cout << foo(v0.begin(), v0.end(), 0.1) << ' ' <<
foo(v1.begin(), v1.end(), 0.1) << std::endl;
return 0;
}

foo函数的返回类型的推导是value_type的推导结果。现在这适用于所有数值类型。

但当value_type是整数类型时，我希望返回类型(以及baz的类型(是double。在这种情况下，我如何进行专业化？

Answer 1

您可以避免专业化或编写另一个重载。相反，您可以使用conditional_t根据某些条件选择特定类型

// alias for convenience
using T = typename std::iterator_traits<ForwIt>::value_type;
// if T is int, value_type is double, otherwise it's just T
using value_type = std::conditional_t<std::is_same_v<T, int>, double, T>;

对于返回类型，只需使用auto，就会从baz的类型中推断出正确的类型。

这是的演示