如何逐一生成行和为v的K×F矩阵C以规避组合爆炸？

这个问题的核心是要避免预计算所有组合导致的内存爆炸，利用partitions包的firstcomposition和nextcomposition逐个生成每个维度的分解，再通过递归/迭代的方式组合成完整的矩阵，而不是一次性生成所有可能的组合。下面我来一步步解释实现思路和代码：

每个符合要求的矩阵C（K行F列）的第f列，本质是将v[f]拆分为K个非负整数的有序和（也就是partitions包中的composition）——因为K个行向量的和是v，所以每一列的元素之和必须等于v[f]。所有合法矩阵的集合，就是各个列的composition的笛卡尔积。但直接生成笛卡尔积会导致组合爆炸，所以我们用逐个遍历+递归组合的方式，每次只处理当前列的一个分解，递归构建完整矩阵，避免一次性存储所有中间结果。

首先加载partitions包：

library(partitions)

1. 逐个处理矩阵的递归函数（无内存存储，适合大组合数） #

这个函数会逐个生成并输出所有合法矩阵，不会把所有矩阵都存在内存里，适合F/K较大的场景：

generate_all_C <- function(v, K, current_cols = list(), col_idx = 1) {
  # 终止条件：所有列都处理完毕，输出当前矩阵
  if (col_idx > length(v)) {
    C <- do.call(cbind, current_cols)
    # 这里可以替换为你需要的逻辑：比如写入文件、计算指标等
    print(C)
    return(invisible(NULL))
  }
  
  # 生成当前列的第一个分解
  current_comp <- firstcomposition(v[col_idx], K)
  while (!is.null(current_comp)) {
    # 递归处理下一列，把当前列的分解加入临时列表
    generate_all_C(v, K, c(current_cols, list(current_comp)), col_idx + 1)
    # 生成当前列的下一个分解
    current_comp <- nextcomposition(current_comp)
  }
}

2. 测试示例（对应题目中的F=2, K=2, v=(1,2)） #

v <- c(1, 2)
K <- 2
generate_all_C(v, K)

运行后会输出6个合法矩阵，和题目描述的结果一致。

3. 可选：将所有矩阵存储到列表中 #

如果需要把所有结果保存下来（注意：组合数过大时会占用大量内存），可以修改函数收集结果：

generate_all_C_list <- function(v, K, current_cols = list(), col_idx = 1, result = list()) {
  if (col_idx > length(v)) {
    C <- do.call(cbind, current_cols)
    result <- c(result, list(C))
    return(result)
  }
  
  current_comp <- firstcomposition(v[col_idx], K)
  while (!is.null(current_comp)) {
    result <- generate_all_C_list(v, K, c(current_cols, list(current_comp)), col_idx + 1, result)
    current_comp <- nextcomposition(current_comp)
  }
  return(result)
}

# 使用示例
all_matrices <- generate_all_C_list(v, K)
length(all_matrices) # 输出6，符合预期

• 避免了expand.grid预计算所有组合的内存开销：我们只在递归过程中临时存储当前正在构建的列，不会一次性生成所有列的分解。
• 完全利用firstcomposition和nextcomposition的迭代特性：逐个生成单个维度的分解，不需要预存该维度的所有分解结果。
• 可以灵活处理大场景：如果组合数极大，你可以在递归的终止条件里直接处理矩阵（比如写入磁盘），而不是存储到内存中。

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