如何使用打字稿制作笛卡尔产品？

Question

这是我所追求的类型签名：

function cartesianProduct<T1, T2, T3, T4, T5, T6, T7, T8>([c1, c2, c3, c4, c5, c6, c7, c8]: [T1[], T2[], T3[], T4[], T5[], T6[], T7[], T8[]]): [T1, T2, T3, T4, T5, T6, T7, T8][];
function cartesianProduct<T1, T2, T3, T4, T5, T6, T7>([c1, c2, c3, c4, c5, c6, c7]: [T1[], T2[], T3[], T4[], T5[], T6[], T7[]]): [T1, T2, T3, T4, T5, T6, T7][];
function cartesianProduct<T1, T2, T3, T4, T5, T6>([c1, c2, c3, c4, c5, c6]: [T1[], T2[], T3[], T4[], T5[], T6[]]): [T1, T2, T3, T4, T5, T6][];
function cartesianProduct<T1, T2, T3, T4, T5>([c1, c2, c3, c4, c5]: [T1[], T2[], T3[], T4[], T5]): [T1, T2, T3, T4, T5][];
function cartesianProduct<T1, T2, T3, T4>([c1, c2, c3, c4]: [T1[], T2[], T3[], T4[]]): [T1, T2, T3, T4][];
function cartesianProduct<T1, T2, T3>([c1, c2, c3]: [T1[], T2[], T3[]]): [T1, T2, T3][];
function cartesianProduct<T1, T2>([c1, c2]: [T1[], T2[]]): [T1, T2][];
function cartesianProduct<T>(sets: T[][]): T[][] {
  // implementation
}

这是我希望它能做的示例。给定：

const input = [
  [ 'a', 'b' ],
  [ 1, 2 ],
];

它会吐出来：

const output = [
  [ 'a', 1 ],
  [ 'a', 2 ],
  [ 'b', 1 ],
  [ 'b', 2 ],
];

尝试使用此参考实现给我们这个类型的问题：

Here's a playground。

Answer 1

如果您想使用ramda，则可以使用 r.SequeSection带有数组。

const cartesianProduct = R.sequence(Array.of)

const input = [
  [ 'a', 'b' ],
  [ 1, 2 ],
];

cartesianProduct(input)
//=> [["a", 1], ["a", 2], ["b", 1], ["b", 2]]

Answer 2

对于此类问题，我一直喜欢首先从类型开始。

首先是提取数组的元素类型的类型：

type ElementType<A> = A extends ReadonlyArray<infer T> ? T : never;

是的，我知道a [number]存在，但这将在以后导致类型错误，因为a将是结果从推断。

然后，一种列出数组列表并为我们提供其元素类型的类型。

type ElementsOfAll<Inputs, R extends ReadonlyArray<unknown> = []> = Inputs extends readonly [infer F, ...infer M] ? ElementsOfAll<M, [...R, ElementType<F>]> : R;

它几乎就像inputs.map（（f）=＆gt; elementType（f））在代码中。

最后，我们使用这种类型来定义cartesianproduct类型：

type CartesianProduct<Inputs> = ElementsOfAll<Inputs>[];

现在我们以类型进行了代码进行此操作：

function cartesianProduct(sets) {
  return sets.reduce((a, b) => a.flatMap(d => b.map(e => [d, e].flat())));
}

这种惊人的实现是从这个出色的答案如果您的环境不支持flat或flatmap，则可以在哪里找到其他替代方案。

在实现此实现之后，我们添加了类型：

function cartesianProduct<Sets extends ReadonlyArray<ReadonlyArray<unknown>>>(sets: Sets): CartesianProduct<Sets> {
  return sets.reduce((a, b) => a.flatMap(d => b.map(e => [d, e].flat())));
}

但是存在错误，因为返回类型与redair调用的类型不匹配。 Unfortunately I don't know a good way to rid the error but if a cast is acceptable:

function cartesianProduct<Sets extends ReadonlyArray<ReadonlyArray<unknown>>>(sets: Sets): CartesianProduct<Sets> {
  return sets.reduce((a, b) => a.flatMap(d => b.map(e => [d, e].flat()))) as CartesianProduct<Sets>;
}

Playground