本文介绍如何在 javascript 中实现带条件限制的笛卡尔积运算,支持基于参数索引与值的嵌套规则(如“当第 0 位为 2 时,第 1 位只能是 'a' 或 'b'”),通过函数式组合与高效过滤完成合规组合生成。
在实际业务中(如动态表单配置、多维筛选器、参数化测试用例生成等),我们常需对多个选项数组求笛卡尔积,但并非所有组合都合法——某些参数值之间存在依赖或互斥关系。例如:“若第一个参数取值为 2,则第二个参数仅允许 'A' 或 'B'”。这类条件性笛卡尔积无法直接用朴素递归或 flatMap 实现,需将约束建模为可复用、可组合的校验逻辑,并与全量组合进行高效过滤。
解决方案采用「先生成后过滤」的清晰分层策略:
const cartesian = ([xs, ...xss]) => {
const kids = xs && cartesian(xss);
return xs == undefined ? [[]] : xs.flatMap(x =>
(kids || [[]]).map(ys => [x, ...ys])
);
};该实现时间复杂度为 O(∏|params[i]|),空间上仅保留当前层级结果,兼顾可读性与性能。
转换代码如下:
立即学习“Java免费学习笔记(深入)”;
const fns = Object.entries(pairRestrictions)
.flatMap(([k0, v0]) =>
Object.entries(v0).flatMap(([k1, v1]) =>
Object.entries(v1).flatMap(([k2, v2]) =>
(xs) => xs[k0] == k1 ? v2.includes(xs[k2]) : true
)
)
);此结构天然支持多级、多条件叠加(如同时限制 0→2→1 和 0→3→2),且每个谓词独立、无副作用。
const generateCombinations = (params, pairRestrictions) => {
const fns = /* 上述转换逻辑 */;
return cartesian(params).filter(comb => fns.every(fn => fn(comb)));
};const params = [[1, 2, 3], ["A", "B", "C"], [10, 11, 12]];
const pairRestrictions = {0: {2: {1
: ["A", "B"]}}};
const result = generateCombinations(params, pairRestrictions);
console.log(result.length); // 24(原始 27 种中排除了 [2,'C',*] 和 [3,'C',*] 共 3 组)输出符合预期:所有 params[0] === 2 或 params[0] === 3 的组合中,params[1] 均不为 "C"。
通过将约束声明式地编码为纯函数,并与通用笛卡尔积解耦,本方案在保持代码简洁性的同时,提供了强大的表达能力与维护性——真正实现「配置即逻辑」。