优雅向量化嵌套for循环：生成合规index结构用于apply函数

你需要构建一个索引结构index，使得apply(index, special_print, MARGIN = 1)的输出和原三重循环完全一致，但当前用expand.grid全量生成再过滤的方法效率极低——30的全笛卡尔积有27万条数据，过滤过程耗时严重。下面给出两种高效解决方案：

方法一：利用多重组合函数（推荐，简洁高效） #

R的gtools包提供了直接生成可重复组合的函数，正好匹配我们需要的i≤j≤k的索引规则：

# 安装并加载gtools（首次使用需安装）
install.packages("gtools")
library(gtools)

vector <- 1:30
# 生成所有满足i≤j≤k的索引组合，顺序与原循环完全一致
index <- combinations(n = length(vector), r = 3, repeats.allowed = TRUE)

验证执行：

apply(index, MARGIN = 1, function(x) special_print(x[1], x[2], x[3]))

效率说明 #

这种方法直接生成符合条件的组合，不需要先构造全量数据再过滤。对于n=30，符合条件的组合数仅为C(30+3-1,3)=4960，远小于27万的全笛卡尔积，速度提升非常显著。

方法二：纯Base R实现（无第三方依赖） #

如果不想依赖gtools包，可以用嵌套的lapply直接生成符合条件的索引矩阵：

vector <- 1:30
n <- length(vector)

# 按i→j→k的顺序生成所有i≤j≤k的组合
index <- do.call(rbind, lapply(1:n, function(i) {
  do.call(rbind, lapply(i:n, function(j) {
    cbind(i, j, k = j:n)
  }))
}))

这种方法同样避免了全量数据的生成，效率远高于你原来的过滤方案，且输出顺序与原循环完全一致。

输出一致性验证 #

两种方法生成的index矩阵，其行顺序完全匹配原三重循环的遍历顺序（i从1到30，j从i到30，k从j到30），因此apply调用special_print后的输出和原循环完全相同。

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