Question

我想问一下，如果你们中的一些人不知道解决这类问题的简单方法：我需要生成从集合B（0,1,2 ... B）中取得的A数的所有组合，其总和= C.

即A=2，B=3，C=2：

在这种情况下的解决方案：

(1,1);(0,2);(2,0)

因此向量是长度2（A），其所有项的总和是2（C），每个向量元素的可能值来自集合{0,1,2,3}（最大值为B）。

Answer 1

z <- expand.grid(0:3,0:3)
z[rowSums(z)==2, ]
  Var1 Var2
3    2    0
5    1    1
7    0    2

如果您想以编程方式执行扩展网格，则可以使用：

z <- expand.grid( rep( list(C), A) )

您需要展开列表，以便项目保持独立。 rep(0:3, 3)不会返回3个单独的序列。所以对于A = 3：

> z <- expand.grid(rep(list(0:3), 3))
> z[rowSums(z)==2, ]
   Var1 Var2 Var3
3     2    0    0
6     1    1    0
9     0    2    0
18    1    0    1
21    0    1    1
33    0    0    2

Answer 2

自从我更新之前已经开始的功能版本：

A=2
B=3
C=2

myfun <- function(a=A, b=B, c=C) {
  out <- do.call(expand.grid, lapply(1:a, function(x) 0:b))
  return(out[rowSums(out)==c,])
}

> out[rowSums(out)==c,]
  Var1 Var2
3    2    0
6    1    1
9    0    2

Answer 3

使用漂亮的partitions()包，以及A，B和C更有趣的值：

library(partitions)

A <- 2
B <- 5
C <- 7

comps <- t(compositions(C, A))
ii <- apply(comps, 1, FUN=function(X) all(X %in% 0:B))
comps[ii, ]
#      [,1] [,2]
# [1,]    5    2
# [2,]    4    3
# [3,]    3    4
# [4,]    2    5

使用R生成向量

3 个答案: