如何高效生成显示二元向量元素对相等性的矩阵？

给定一个仅含0和1的向量，例如：

V1 <- c(1,0,1,1,0)

需要生成一个与向量长度相同的方阵M，满足当向量中第i个和第j个元素相等时，M[i,j] = 1，否则为0。上述示例对应的矩阵为：

[,1] [,2] [,3] [,4] [,5]
[1,]    1    0    1    1    0
[2,]    0    1    0    0    1
[3,]    1    0    1    1    0
[4,]    1    0    1    1    0
[5,]    0    1    0    0    1

由于实际场景中向量长度为44，且需要重复模拟约10000次，for循环效率极低，因此需要高效的向量化实现方案。

高效实现方法 #

以下几种方法均为向量化操作，避免了循环带来的性能损耗，适合大规模重复调用：

方法1：使用outer函数直接比较

outer函数可对两个向量的所有元素对执行指定操作，这里直接用相等比较，再转换为整数类型：

generate_eq_matrix <- function(v) {
  as.integer(outer(v, v, "==")) %>% matrix(nrow = length(v))
}

# 测试示例
V1 <- c(1,0,1,1,0)
M1 <- generate_eq_matrix(V1)
print(M1)

输出结果与示例完全匹配。

方法2：利用矩阵乘法（更高效的数值运算）

由于向量仅含0和1，可通过矩阵乘法直接计算：
当V[i]和V[j]同为1时，V[i]*V[j] = 1；同为0时，(1-V[i])*(1-V[j]) = 1，两者相加即可得到目标矩阵：

generate_eq_matrix <- function(v) {
  v_mat <- matrix(v, ncol = 1)
  (v_mat %*% t(v_mat)) + ((1 - v_mat) %*% t(1 - v_mat))
}

# 测试示例
M2 <- generate_eq_matrix(V1)
print(M2)

此方法完全基于数值矩阵运算，在大规模重复调用时性能略优于outer。

性能说明 #

两种方法均为向量化操作，时间复杂度为O(n²)但避免了循环的常数开销。对于44长度的向量，单次生成矩阵的时间可忽略不计，10000次模拟的总耗时远低于for循环方案。

内容的提问来源于stack exchange，提问作者Paul Mee