如何在MATLAB中自动实现多层for循环？含向量化方案需求

一、问题本质分析 #

你的代码实际是计算满足 (i_1 + i_2 + ... + i_n \leq k) 的非负整数解的个数（示例中n=4）。这个问题无需嵌套循环，直接用组合数学公式就能高效求解，不过还是针对你的两个需求分别给出方案：

二、自动扩展至任意层数（如100层）的循环实现 #

如果一定要保留循环逻辑，可以用递归函数避免手动写100层嵌套：

function total = countSolutions(k, n)
    total = 0;
    recurse(0, n, k);
    
    function recurse(currentSum, remainingVars, maxSum)
        if remainingVars == 0
            total = total + 1;
            return;
        end
        for i = 0 : maxSum - currentSum
            recurse(currentSum + i, remainingVars - 1, maxSum);
        end
    end
end

调用方式：

k = 100;
n = 100; % 对应i₁到i₁₀₀
count = countSolutions(k, n);

⚠️ 注意：当n=100、k=100时，递归计算量极大，运行会极慢甚至内存溢出，不建议实际使用。

三、向量化/公式化最优方案 #

这个问题属于可重复组合问题，满足 (i_1 + i_2 + ... + i_n \leq k) 的非负整数解个数，等价于引入松弛变量 (i_{n+1}) 后 (i_1 + i_2 + ... + i_n + i_{n+1} = k) 的解个数，根据隔板法公式，结果为：
[
\binom{k + n}{n}
]

在MATLAB中直接用nchoosek计算，数值过大时用高精度运算避免溢出：

k = 100;
n = 100; % 变量个数
count = nchoosek(k + n, n);
% 高精度版本
count_vpa = vpa(nchoosek(sym(k + n), sym(n)));

该方案无循环，计算效率极高，哪怕n=100、k=100也能瞬间出结果。

验证示例（n=4，k=100） #

原循环代码的计算结果与nchoosek(100+4,4)=4598126完全一致，证明公式正确性。

内容的提问来源于stack exchange，提问作者Liszt Morero