从模板包生成大小为 N 的所有子包

Generating all subpacks of size N from a template pack

本文关键字:小为 包生成      更新时间:2023-10-16
NSubsets<N, Pack<Types...>>::type是由

大小为 N 的所有Types...子集组成的包包。例如

NSubsets<2, Pack<int, char, double>>::type

是要

Pack<Pack<int, char>, Pack<int, double>, Pack<char, double>>

一种方法是简单地从从包中获取所有子包中获取PowerSet解决方案的输出,然后取出每个不是 N 码的包装。 但是这对于大N来说效率太低了(无论如何都是糟糕的(。 这是我的想法(灵感来自优雅的PowerSet解决方案(: 假设我们有 Pack<A,B,C,D> 个,并且 N = 2。 从 Pack<> 开始,我们遍历 Pack<A,B,C,D> 中的类型并附加每个类型,如下所示:在附加任何内容之前,我们有:

Pack<>

将 A 附加到前一个(并保留前一个(,我们得到:

Pack<>, Pack<A>

将 B 附加到前一个(并保留前一个(,我们得到:

Pack<>, Pack<A>, Pack<B>, Pack<A,B>,

但是Pack<A,B>有 2 号,所以把它藏起来,从这个列表中拿出来,给我们留下:

Pack<>, Pack<A>, Pack<B>

将 C 附加到前一个(并保留前一个(,我们得到:

Pack<>, Pack<A>, Pack<B>, Pack<C>, Pack<A,C>, Pack<B,C>.

像上面一样,藏匿Pack<A,C>, Pack<B,C>

Pack<>, Pack<A>, Pack<B>, Pack<C>

将 D 附加到前一个(并保留前一个(,我们得到:

Pack<D>, Pack<A,D>, Pack<B,D>, Pack<C,D>.

再次拿 2 码的那些,我们现在终于有了

Pack<Pack<A,B>, Pack<A,C>, Pack<B,C>, Pack<A,D>, Pack<B,D>, Pack<C,D>>

作为我们想要的输出。

请注意,此算法中的一个缺陷是无缘无故地将Pack<>保持在倒数第二步。 如果 N 大于 2,这个无关的部分真的会浪费时间。 以下是我使用上述方法的代码,但输出给出 false,我无法追踪原因(还(。 但即使如果它确实正常工作,我仍然不太喜欢它,主要是因为我刚才提到的上面提到的缺陷,我不知道如何以消除该缺陷。

#include <iostream>
#include <type_traits>
template <int, typename> struct IsSize;
template <int N, template <typename...> class P, typename... Types>
struct IsSize<N, P<Types...>> : std::integral_constant<bool, sizeof...(Types) == N> {};
template <int, typename, typename, typename> struct PartitionPacksBySizeHelper;
template <int N, template <typename...> class P, typename... KeptPacks, typename... SizeNPacks>
struct PartitionPacksBySizeHelper<N, P<>, P<KeptPacks...>, P<SizeNPacks...>> {
    using not_sizeN_types = P<KeptPacks...>;
    using sizeN_types = P<SizeNPacks...>;
};
template <int N, template <typename...> class P, typename First, typename... Rest, typename... KeptPacks, typename... SizeNPacks>
struct PartitionPacksBySizeHelper<N, P<First, Rest...>, P<KeptPacks...>, P<SizeNPacks...>> : std::conditional<IsSize<N, First>::value,
        PartitionPacksBySizeHelper<N, P<Rest...>, P<KeptPacks...>, P<SizeNPacks..., First>>,
        PartitionPacksBySizeHelper<N, P<Rest...>, P<KeptPacks..., First>, P<SizeNPacks...>>
    >::type {};
template <int, typename> struct PartitionPacksBySize;
template <int N, template <typename...> class P, typename... Packs>
struct PartitionPacksBySize<N, P<Packs...>> : PartitionPacksBySizeHelper<N, P<Packs...>, P<>, P<>> {};
template <typename, typename> struct Append;
template <typename T, template <typename...> class P, typename...Types>
struct Append<T, P<Types...>> {
    using type = P<Types..., T>;
};
template <int, typename, typename, typename> struct NSubsetsHelper;
template <int N, template <typename...> class P, typename... CurrentPacks, typename... AccumulatedPacks>
struct NSubsetsHelper<N, P<>, P<CurrentPacks...>, P<AccumulatedPacks...>> {
    using type = P<AccumulatedPacks...>;
};
template <int N, template <typename...> class P, typename First, typename... Rest, typename... KeptPacks, typename... SizeNPacks>
struct NSubsetsHelper<N, P<First, Rest...>, P<KeptPacks...>, P<SizeNPacks...>>
    : NSubsetsHelper<N, P<Rest...>,
        typename PartitionPacksBySize<N, P<KeptPacks..., typename Append<First, KeptPacks>::type...>>::not_sizeN_types,
        typename PartitionPacksBySize<N, P<KeptPacks..., typename Append<First, KeptPacks>::type...>>::sizeN_types> {};
template <int, typename> struct NSubsets;
template <int N, template <typename...> class P, typename...Types>
struct NSubsets<N, P<Types...>> : NSubsetsHelper<N, P<Types...>, P<P<>>, P<>> {};
// -----------------------------------------------------------------------------------------------------------------------------------------------
// Testing
template <typename...> struct Pack {};
int main() {
    std::cout << std::boolalpha << std::is_same< NSubsets<2, Pack<int, char, double>>::type,
        Pack<Pack<int, char>, Pack<int, double>, Pack<char, double>>
    >::value << std::endl;  // false (darn!)
}

我在纸上追踪到上面的包应该是输出,当我改变包的顺序时,它仍然给出 false。 但是,正如我上面提到的,该方法无论如何都是劣质的。 关于更好的方法有什么建议吗?

更新:我发现了我的错误,并更换了

typename PartitionPacksBySize<N, P<KeptPacks..., typename Append<First, KeptPacks>::type...>>::sizeN_types>

typename Merge<P<SizeNPacks...>, typename PartitionPacksBySize<N, P<KeptPacks..., typename Append<First, KeptPacks>::type...>>::sizeN_types>::type

但是,你仍然看到我的算法在 N 很大时的最后 N 次迭代中是如何浪费时间的?

我们可以从一开始就精确地生成大小k子集 - 这将更有效,因为我们只需要做O(n^k)工作而不是O(2^n)工作。这里的算法只是迭代具有正好k 1 的 n 位单词的所有排列,并为每个单词添加适当的Pack

我们首先从一个 bithack 开始,以 constexpr 的形式查找下一个排列:

constexpr int ctz(size_t n) {
    return n & 1 ? 0 : 1 + ctz(n >> 1); 
}
constexpr size_t next_perm_impl(size_t v, size_t t) {
    return (t + 1) | (((~t & -~t) - 1) >> (ctz(v) + 1));
}
constexpr size_t next_perm(size_t v) {
    return next_perm_impl(v, v | (v - 1));
}

接下来,我抓住哥伦布的累加器,他出于某种原因从你之前问题的答案中删除了它:

template <class... T> struct Pack { using type = Pack; };
template <size_t size, class result, class>
struct accumulator : result { };
template <size_t j, class... R, class T1, class... T>
struct accumulator<j, Pack<R...>, Pack<T1, T...>>
: accumulator<(j>>1), typename std::conditional<j&1, Pack<R..., T1>, Pack<R...>>::type, Pack<T...>>
{};

给定一个值j,我们可以确定与这些元素相关联的Pack。现在我们只需要从(1 << k) - 1迭代到(1 << N) + (1 << (k-1)) - 1 .可能有一种更有效的方法可以做到这一点,但以下工作:

template <typename P, typename Result, size_t CUR, size_t LAST>
struct PowerPackImpl;
template <typename P, typename... R, size_t CUR, size_t LAST>
struct PowerPackImpl<P, Pack<R...>, CUR, LAST>
: PowerPackImpl<P,
                Pack<R..., typename accumulator<CUR, Pack<>, P>::type>,
                next_perm(CUR),
                LAST>
{ };
template <typename P, typename... R, size_t LAST>
struct PowerPackImpl<P, Pack<R...>, LAST, LAST>
: Pack<R...> { };
template <typename P, size_t K> struct PowerPack;
template <typename... P, size_t K>
struct PowerPack<Pack<P...>, K>
: PowerPackImpl<Pack<P...>, Pack<>, (1 << K) - 1, (1 << sizeof...(P)) + (1 << (K-1)) - 1>
{ };

例如:

static_assert(std::is_same<
    typename PowerPack<Pack<int, char, double, float>, 1>::type,
    Pack<Pack<int>, Pack<char>, Pack<double>, Pack<float>>
>::value, "1 works");
static_assert(std::is_same<
    typename PowerPack<Pack<int, char, double, float>, 2>::type,
    Pack<Pack<int, char>, Pack<int, double>, Pack<char, double>, Pack<int, float>, Pack<char, float>, Pack<double, float> >
>::value, "2 works");

这是基于PowerPack解决方案,但过长条目的过滤发生在每个步骤中,因此比仅在最后过滤更有效。

步骤1:NAppend<N,Pack<...>,T>仅将新类型T附加到Pack<...>,如果之后的条目不超过N

template<std::size_t,typename,typename>
struct NAppend
{
    using type = void;
};
template<std::size_t N,typename...Ts,typename T>
struct NAppend<N,Pack<Ts...>,T>
{
    using type =
        typename std::conditional<
            sizeof...(Ts)==N,
            void,
            Pack<Ts...,T>
        >::type;
};

第 2 步:ShrinkPackPack s 和 void s 的Pack中删除void

template<typename,typename U=Pack<>>
struct ShrinkPack
{
    using type = U;
};
template<typename T,typename...Ts,typename...Us>
struct ShrinkPack<Pack<T,Ts...>,Pack<Us...>>
    : std::conditional<
        std::is_void<T>::value,
        ShrinkPack<Pack<Ts...>,Pack<Us...>>,
        ShrinkPack<Pack<Ts...>,Pack<Us...,T>>
      >::type
{
};

步骤3:NPack过滤掉所有大小不正确的条目

template<std::size_t,typename,typename U=Pack<>>
struct NPack
{
    using type = U;
};
template<std::size_t N,typename...Ts,typename...Us,typename...Vs>
struct NPack<N,Pack<Pack<Ts...>,Us...>,Pack<Vs...>>
    : std::conditional<
        sizeof...(Ts)==N,
        NPack<N,Pack<Us...>,Pack<Vs...,Pack<Ts...>>>,
        NPack<N,Pack<Us...>,Pack<Vs...>>
      >::type
{
};

最后一步:适应扩展,类似于PowerPack,在每个扩展步骤应用ShrinkPack,最后NPack一次。

template<std::size_t N,typename,typename T=Pack<Pack<>>>
struct NPowerPack
{
    using type = typename NPack<N,T>::type;
};
template<std::size_t N,typename T,typename...Ts,typename...Us>
struct NPowerPack<N,Pack<T,Ts...>,Pack<Us...>>
    : NPowerPack<N,Pack<Ts...>,typename ShrinkPack<Pack<Us...,typename NAppend<N,Us,T>::type...>>::type>
{
};

测试:

static_assert(std::is_same<
    NPowerPack<1,Pack<int, char, double>>::type,
    Pack<Pack<int>, Pack<char>, Pack<double>>
>(), "");
static_assert(std::is_same<
    NPowerPack<2,Pack<int, char, double>>::type,
    Pack<Pack<int, char>, Pack<int, double>, Pack<char, double>>
>(), "");

现场示例