所有N个所有子集的组合

时间:2015-02-05 20:50:21

标签: r combinatorics

给定元素向量,我想获得所有可能的n长度元素子集组合的列表。例如,给定(最简单的)序列1:2,我想获得表单的列表对象

{ {{1},{1}}, {{1},{2}}, {{2},{2}}, {{1},{1,2}}, {{2},{1,2}}, {{1,2},{1,2}} }

n=2

我能够使用以下内容生成所有非空子集的列表:

listOfAllSubsets <- function (s) {
  n <- length(s)
  unlist(lapply(1:n, function (n) {
    combn(s, n, simplify=FALSE)
  }), recursive=FALSE)
}

但是,我不确定从这里开始的最佳方式。基本上,我想要一个这个列表的笛卡尔积(对于n=2)。

有什么建议吗?非迭代解决方案将是更可取的(即,没有for循环)。

3 个答案:

答案 0 :(得分:3)

从索引的笛卡尔积开始更容易。然后通过确保索引的元组被排序来避免重复。

combosn <- function(items,n) {
  i <- seq_along(items)
  idx <-do.call(expand.grid,rep(list(i),n))
  idx <- idx[!apply(idx,1,is.unsorted),]
  apply(idx,1,function(x) items[x])
}

ss<-listOfAllSubsets(1:2)

str(combosn(ss,2))
List of 6
 $ :List of 2
  ..$ : int 1
  ..$ : int 1
 $ :List of 2
  ..$ : int 1
  ..$ : int 2
 $ :List of 2
  ..$ : int 2
  ..$ : int 2
 $ :List of 2
  ..$ : int 1
  ..$ : int [1:2] 1 2
 $ :List of 2
  ..$ : int 2
  ..$ : int [1:2] 1 2
 $ :List of 2
  ..$ : int [1:2] 1 2
  ..$ : int [1:2] 1 2

或者,对于n=3

str(combosn(ss,3))
List of 10
 $ :List of 3
  ..$ : int 1
  ..$ : int 1
  ..$ : int 1
 $ :List of 3
  ..$ : int 1
  ..$ : int 1
  ..$ : int 2
 $ :List of 3
  ..$ : int 1
  ..$ : int 2
  ..$ : int 2
 $ :List of 3
  ..$ : int 2
  ..$ : int 2
  ..$ : int 2
 $ :List of 3
  ..$ : int 1
  ..$ : int 1
  ..$ : int [1:2] 1 2
 $ :List of 3
  ..$ : int 1
  ..$ : int 2
  ..$ : int [1:2] 1 2
 $ :List of 3
  ..$ : int 2
  ..$ : int 2
  ..$ : int [1:2] 1 2
 $ :List of 3
  ..$ : int 1
  ..$ : int [1:2] 1 2
  ..$ : int [1:2] 1 2
 $ :List of 3
  ..$ : int 2
  ..$ : int [1:2] 1 2
  ..$ : int [1:2] 1 2
 $ :List of 3
  ..$ : int [1:2] 1 2
  ..$ : int [1:2] 1 2
  ..$ : int [1:2] 1 2

答案 1 :(得分:2)

这就是我要做的事情,例如,s=1:2

1)为每个元素的成员资格表示具有0/1矩阵的子集。

subsets = as.matrix(do.call(expand.grid,replicate(length(s),0:1,simplify=FALSE)))

给出了

     Var1 Var2
[1,]    0    0
[2,]    1    0
[3,]    0    1
[4,]    1    1

这里,第一行是空子集;第二,{1};第三,{2};第四,{1,2}。要获取子集本身,请使用mysubset = s[subsets[row,]],其中row是您想要的子集行。

2)将子集对表示为矩阵的行对:

pairs <- expand.grid(Row1=1:nrow(subsets),Row2=1:nrow(subsets))

给出了

   Row1 Row2
1     1    1
2     2    1
3     3    1
4     4    1
5     1    2
6     2    2
7     3    2
8     4    2
9     1    3
10    2    3
11    3    3
12    4    3
13    1    4
14    2    4
15    3    4
16    4    4

这里,第十四行对应于subsets的第二和第四行,因此{1}&amp; {1,2}。这假设对的顺序很重要(这在采用笛卡尔积时是隐含的)。要恢复子集,请使用mypairosubsets=lapply(pairs[p,],function(r) s[subsets[r,]]),其中p是您想要的对的行。

将对数扩展到P(s)^n案例(其中P(s)s的权力集)看起来像

setsosets = as.matrix(do.call(expand.grid,replicate(n,1:nrow(subsets),simplify=FALSE)))

这里,每行都有一个数字向量。每个数字对应subsets矩阵中的一行。


制作s元素的副本可能不需要在此之后做的任何事情。但是,你可以使用lapply(1:nrow(pairs),function(p)lapply(pairs[p,],function(r) s[subsets[r,]]))从这里开始,它的开始就像......

[[1]]
[[1]]$Row1
integer(0)

[[1]]$Row2
integer(0)


[[2]]
[[2]]$Row1
[1] 1

[[2]]$Row2
integer(0)

答案 2 :(得分:1)

allSubsets<-function(n,# size of initial set
                     m,# number of subsets
                    includeEmpty=FALSE)# should the empty set be consiered a subset?

{

    # m can't exceed the number of possible subsets
    if(includeEmpty)
        stopifnot(m <= 2^n)
    else
        stopifnot(m <= 2^n-1)

    # get the subsets of the initial set (of size n)
    if(includeEmpty){
        ll <- split(t(combn(2^n,m)),seq(choose(2^n,m)))
    }else
        ll <- split(t(combn(2^n-1,m)),seq(choose(2^n-1,m)))

    # get the subets 
    subsets <- apply(do.call(expand.grid,rep(list(c(F,T)),n)),
                     1,which)
    # remove the empty subset if desired
    if(!includeEmpty)
        subsets <- subsets[-1]

    # covert the subsets to vector
    subsets <- lapply(subsets,as.vector)

    # return the list of subsets
    apply(t(mapply('[',list(subsets),ll)),1,function(x)x)

}

# returns a list where each element is a list of length 2 with 
# subsets of the initial set of length 4
x = allSubsets(4,2,F)