
本文介绍如何在 javascript 中实现通用的算术-几何平均(agm)单步迭代函数,支持任意长度的正实数数组,通过位运算高效生成初等对称多项式并结合二项式系数归一化,最终输出下一轮 agm 序列。
本文介绍如何在 javascript 中实现通用的算术-几何平均(agm)单步迭代函数,支持任意长度的正实数数组,通过位运算高效生成初等对称多项式并结合二项式系数归一化,最终输出下一轮 agm 序列。
算术-几何平均(AGM)的经典定义针对两个数:反复交替计算算术平均与几何平均,序列收敛于同一极限。其高维推广基于初等对称多项式(elementary symmetric polynomials) 的规范化幂平均——对长度为 $ n $ 的正实数数组 $ [x_0, x1, \dots, x{n-1}] $,第 $ k $ 个 AGM 分量($ k = 0, 1, \dots, n-1 $)定义为:
$$ a_k = \left( \frac{e_k(x0,\dots,x{n-1})}{\binom{n}{k}} \right)^{1/k}, \quad \text{其中 } e_k \text{ 是 } k\text{-阶初等对称和,且约定 } a_0 = \frac{e1}{n} \text{(算术平均),} a{n-1} = (x_0 x1 \cdots x{n-1})^{1/n} \text{(几何平均)。} $$
关键在于:对每个 $ k \in [1, n-1] $,需枚举所有 $ \binom{n}{k} $ 个大小为 $ k $ 的子集,计算其元素乘积之和 $ e_k $,再除以组合数 $ \binom{n}{k} $,最后开 $ k $ 次方根。
手动展开(如 agm3、agm6)不可扩展。高效通用解法是利用位掩码遍历幂集:对 $ 1 $ 到 $ 2^n - 1 $ 的每个整数 $ i $,将其二进制位视为子集选择信号(第 $ b $ 位为 1 表示选取 $ x_b $),统计置位数 $ k $,累加对应乘积到 sums[k]。该方法时间复杂度 $ O(n \cdot 2^n) $,适用于中小规模($ n \leq 20 $);数值稳定前提下,对典型用例($ n \leq 10 $)性能优异。
以下是完整、健壮、可直接使用的通用 agm 函数实现:
function agm(arr) {
if (!Array.isArray(arr) || arr.length === 0) {
throw new Error('Input must be a non-empty array');
}
const n = arr.length;
// 验证全为严格正实数
for (let i = 0; i 0)
result[k] = Math.pow(avg, 1 / k);
}
}
return result;
}
// 辅助函数:计算组合数 C(n, k),使用迭代避免大数阶乘溢出
function binomial(n, k) {
if (k n) return 0;
if (k === 0 || k === n) return 1;
k = Math.min(k, n - k); // 利用对称性优化
let res = 1;
for (let i = 0; i <p><strong>注意事项与最佳实践:</strong></p>
- ✅ 输入校验:函数强制检查数组非空、元素为严格正有限数,避免
NaN或发散; - ✅ 数值稳定性:使用
Math.pow(..., 1/k)而非Math.exp(Math.log(...)/k),兼顾精度与可读性;对binomial使用迭代计算,规避大阶乘溢出; - ⚠️ 性能边界:位枚举法复杂度为 $ O(n \cdot 2^n) $,$ n > 20 $ 时计算显著变慢,生产环境建议限制输入长度(如
n ≤ 16),或对超大规模采用近似算法(如随机采样子集); - ? 迭代收敛:单次
agm()仅完成一步变换;要获得 AGM 极限值,需循环调用直至各分量差异小于容差(例如max(|a_i - a_j|) ); - ? 数学一致性:本实现严格遵循高维 AGM 定义——$ a0 $ 为算术平均,$ a{n-1} $ 为几何平均,中间项为对称均值的幂平均形式。
通过此通用实现,开发者无需为不同维度重复编写逻辑,即可无缝集成高维 AGM 计算至数值分析、椭圆积分近似或特殊函数库中。










