如何在R中编写函数验证细致平衡方程是否成立？

我们需要编写一个R函数，接收任意m×m的转移矩阵和长度为m的概率向量作为参数，验证所有细致平衡方程 p_i * P_ij = p_j * P_ji 是否成立：

• 若所有方程都满足，返回字符串 "balanced"
• 若存在不满足的情况，返回 "not_balanced"

示例输入数据 #

题目中的示例出自Robert Dobrow所著《Introduction to stochastic processes in R》，对应的R代码如下：

# 3×3转移矩阵P
P = matrix(c(0,2/5,3/5,
             1/2,1/4,1/4,
             1/2,1/6,1/3),3,3,byrow = TRUE)
P

# 概率向量p
p = c(1/3,4/15,2/5)

函数实现 #

check_detailed_balance <- function(P, p) {
  # 校验输入维度合法性
  m <- nrow(P)
  if (ncol(P) != m || length(p) != m) {
    stop("输入的矩阵必须是方阵，且向量长度需与矩阵维度一致")
  }
  
  # 构造细致平衡方程左右两边的矩阵
  left_mat <- outer(p, rep(1, m)) * P
  right_mat <- outer(rep(1, m), p) * t(P)
  
  # 考虑浮点数精度，检查所有元素是否近似相等
  all_equal <- all(abs(left_mat - right_mat) < 1e-10)
  
  if (all_equal) {
    "balanced"
  } else {
    "not_balanced"
  }
}

关键逻辑说明 #

• 输入校验：先确保矩阵是方阵，且向量长度与矩阵维度匹配，避免后续计算出错
• 矩阵构造：利用outer函数快速生成左右两边的对比矩阵，左边是每个状态的概率p_i乘以转移概率P_ij，右边是目标状态的概率p_j乘以反向转移概率P_ji
• 精度处理：由于浮点数计算存在微小误差，不直接用==判断相等，而是检查差值绝对值是否小于极小阈值（1e-10）
• 结果返回：根据所有元素是否满足条件，返回对应的状态字符串

测试示例 #

运行以下代码测试示例数据：

check_detailed_balance(P, p)

返回结果为"balanced"，符合预期。

内容的提问来源于stack exchange，提问作者Homer Jay Simpson