Matlab递归函数生成元胞数组元素全组合的问题排查

问题说明 #

给定由向量组成的元胞数组，比如：

C = {[1 2 3 4 5], [6:13], [14 15]}

需要生成所有可能的元素组合——每个组合从每个子向量中各取一个元素，输出为元胞数组，形式如下：

out = {[1 6 14], [1 6 15], [1 7 14], ..., [5 13 15]}

原代码问题分析 #

你尝试的递归函数存在以下问题：

• 多余的外层循环k=i:N，导致逻辑混乱，我们只需要按顺序遍历每个元胞的元素，不需要跨元胞循环
• 输出拼接错误：直接将所有元素合并到一个数组，没有生成每个组合作为独立元胞
• 递归参数传递错误：i作为递归深度标记，被修改后返回会打乱后续循环的执行逻辑

正确递归实现 #

function out = allCombinations(cells)
    if isempty(cells)
        out = {};
        return;
    end
    % 处理最后一个元胞，每个元素单独作为初始组合
    if length(cells) == 1
        out = arrayfun(@(x) [x], cells{1}, 'UniformOutput', false);
        return;
    end
    % 递归获取后续元胞的所有组合
    restCombos = allCombinations(cells(2:end));
    % 拼接当前元胞元素与后续组合
    out = {};
    currentVec = cells{1};
    for val = currentVec
        for combo = restCombos
            out = [out, {[val, combo{1}]}];
        end
    end
end

使用方法 #

调用时直接传入目标元胞数组即可：

C = {[1 2 3 4 5], [6:13], [14 15]};
out = allCombinations(C);

高效非递归实现（基于ndgrid） #

如果元胞数组规模较大，递归可能效率有限，可以使用ndgrid实现更高效的非递归版本：

function out = allCombinationsNonRecursive(cells)
    % 生成网格索引，覆盖所有元素组合
    grids = cell(1, length(cells));
    [grids{:}] = ndgrid(cells{:});
    % 将网格数据拼接为统一矩阵
    combined = cat(length(cells)+1, grids{:});
    % 转换为目标元胞数组格式
    out = mat2cell(combined, ones(size(combined,1),1), length(cells));
end

内容的提问来源于stack exchange，提问作者Daisuke Takeshita