具有两个以上参数的变量模板函数

时间:2016-03-15 08:29:41

标签: c++ recursion variadic-templates

我有以下示例,其中使用了两个参数t1和t2。

template<typename T>
bool Compare(T t1, T t2)
{
    return t1 == t2;
}

template<typename T, typename... Args> 
bool Compare(T t1, T t2, Args...  args)
{
    return (t1 == t2) && Compare(args...);
}

int main(void)
{
    Compare(1, 1, "string", "string");
}

函数比较采用相同类型且可以比较的参数对。 比较两对,然后递归地传递参数包,直到达到最后两个参数。 为了停止递归,我使用了不带参数包的比较函数的实现。

我想添加第三个参数t3,因此函数Compare应该是这样的:

template<typename T>
bool Compare(T t1, T t2, T t3)
{
    return t1 == t2 == t3;
}

template<typename T, typename... Args>
bool Compare(T t1, T t2, T t3, Args... args)
{
    return (t1 == t2 == t3) && Compare(args...);
}

int main(void)
{
    Compare(1, 1, 1, "string", "string", "string");
}

我希望这个函数需要三个参数进行比较,然后接下来的三个参数被递归处理。 当我尝试编译此代码时,我得到以下错误:

>xxx\source.cpp(4): error C2446: '==': no conversion from 'const char *' to 'int'
1>  xxx\source.cpp(4): note: There is no context in which this conversion is possible
1>  xxx\source.cpp(10): note: see reference to function template instantiation 'bool Compare<const char*>(T,T,T)' being compiled
1>          with
1>          [
1>              T=const char *
1>          ]
1>  xxx\source.cpp(15): note: see reference to function template instantiation 'bool Compare<int,const char*,const char*,const char*>(T,T,T,const char *,const char *,const char *)' being compiled
1>          with
1>          [
1>              T=int
1>          ]
1>xxx\source.cpp(4): error C2040: '==': 'int' differs in levels of indirection from 'const char *'

如何实现此功能来比较同一类型的三个参数的集合?

2 个答案:

答案 0 :(得分:8)

t1 == t2 == t3

这不会检查t1t2t3是否相等,它会检查t1是否等于t2,然后检查是否结果bool等于t3

也许你想要的是(假设合理的相等运算符):

t1 == t2 && t1 == t3

所以你的代码看起来像这样:

template<typename T>
bool Compare(T t1, T t2, T t3)
{
    return t1 == t2 && t1 == t3;
}

template<typename T, typename... Args>
bool Compare(T t1, T t2, T t3, Args... args)
{
    return t1 == t2 && t1 == t3 && Compare(args...);
}

请注意,使用字符串文字进行的测试调用会进行指针比较,这可能不是您想要的。

答案 1 :(得分:0)

这是我更通用的解决方案。只有当所有N个参数的N个连续参数都相等时,compareConsecutive<N>才会返回true。

#include <iostream>
#include <tuple>
#include <utility>

template <std::size_t Start, typename IndexSequence> struct MakeIndexSequenceHelper;

template <std::size_t Start, std::size_t... Is>
struct MakeIndexSequenceHelper<Start, std::index_sequence<Is...>> {
    using type = std::index_sequence<(Start + Is)...>;
};

template <std::size_t Start, std::size_t Length>
struct MakeIndexSequence : MakeIndexSequenceHelper<Start, std::make_index_sequence<Length>> {};

template <typename T, typename U>
bool allAreSame (const T&, const U&) {
    return false;
}

template <typename T>
bool allAreSame (const T& t1, const T& t2) {
    return t1 == t2;
}

template <typename T, typename U, typename... Args>
bool allAreSame (const T&, const U&, const Args&...) {
    return false;
}

template <typename T, typename... Args>
bool allAreSame (const T& t1, const T& t2, const Args&... args) {
    return allAreSame(t1, t2) && allAreSame(t1, args...);
}

template <typename Tuple, std::size_t... Is>
bool allAreSameHelper (Tuple&& tuple, std::index_sequence<Is...>) {
    return allAreSame (std::get<Is>(std::forward<Tuple>(tuple))...);
}

template <std::size_t N, typename... Args>
bool allAreSameHelper (Args&&... args) {
    return allAreSameHelper (std::forward_as_tuple(std::forward<Args>(args)...), std::make_index_sequence<N>{});
}

template <std::size_t N, typename... Args> bool compareConsecutive (Args&&...);

template <std::size_t N>
bool compareConsecutive() {return true;}

template <std::size_t N, typename Tuple, std::size_t... Is>
bool compareConsecutiveHelper (Tuple&& tuple, std::index_sequence<Is...>) {
    return compareConsecutive<N> (std::get<Is>(std::forward<Tuple>(tuple))...);
}

template <std::size_t N, std::size_t Start, std::size_t Length, typename... Args>
bool compareConsecutiveHelper (Args&&... args) {
    return compareConsecutiveHelper<N> (std::forward_as_tuple(std::forward<Args>(args)...), typename MakeIndexSequence<Start, Length>::type{});
}

template <std::size_t N, typename... Args> 
bool compareConsecutive (Args&&... args) {
    return allAreSameHelper<N>(std::forward<Args>(args)...) && compareConsecutiveHelper<N, N, sizeof...(Args) - N>(args...);
}

int main() {
    std::cout << std::boolalpha << allAreSame("hi", "hi", "hi", "hi", "hi", "hi") << '\n';  // true
    std::cout << compareConsecutive<2>(1, 1, "hi", "hi") << '\n';  // true
    std::cout << compareConsecutive<2>(1, "hi", "hi", "hi") << '\n';  // false
    std::cout << compareConsecutive<3>(1, 1, 1, "hi", "hi", "hi", 4.5, 4.5, 4.5) << '\n';  // true
    std::cout << compareConsecutive<5>(1, 1, 1, 1, 1, "hi", "hi", "hi", "hi", "hi") << '\n';  // true
    std::cout << compareConsecutive<5>(1, 1, 1, 1, 2, "hi", "hi", "hi", "hi", "hi") << '\n';  // false
}