我对从递归函数返回值有疑问,希望你能帮我。我有一个包含一堆矩阵的列表,每个矩阵代表一组可能的组合,并使用compn()生成。例如,列表中可能有3个矩阵:
# set 1 has 4 elements, do nCk = 4C1:
set1 <- c("b2","b3","b4","b5")
set1 <- combn(set1,1,simplify = T)
# set 2 has 3 elements, choose 2:
set2 <- c("c1","c2","b2")
set2 <- combn(set2,2,simplify = T)
# set 3 has 10 elements, choose 1:
set3 <- combn(c(1:10),1, simplify = T)
例如,如果我们要打印set2,它将有2行(选择2)和3列(3C2=3):
> set2
[,1] [,2] [,3]
[1,] "c1" "c1" "c2"
[2,] "c2" "b2" "b2"
我需要得到所有可能的4元素组合(上面的每组1个元素)。我可以使用while循环和模拟状态机来做到这一点,但是这个解决方案很笨重,并且需要很长的代码。我知道这可以使用递归来完成,因为我能够正确打印120个组合(下面的代码),但是当尝试返回它们或将它们保存在变量中时,要么我得到一个
当然,当部署时,集合的数量将是动态的,每个集合的元素也会改变。集合也不太大,所以我很想实现递归方法!
要打印的工作代码:
combb <- function(allsets, number, carry){
if(number>length(allsets)){
print(carry)
return()
} else{
for(j in 1:length(allsets[[number]][1,])){
newcarry <- c(carry, allsets[[number]][,j])
number2 <- number + 1
combb(allsets, number2, newcarry)
}
}
}
谢谢你!
我发现来回携带结果非常困难,因为它需要标志和列表或不同的解决方案。我所做的是创建一个包装函数,在其中创建一个局部变量。递归函数在内部定义,并访问(“全局”)上面提到的变量。然后,该变量由包装器返回:
combb <- function(allsets){
carry <- integer(0)
height <- 0L
for (j in 1:length(allsets)) {
height <- height + length(allsets[[j]][, 1])
}
output <- matrix(allsets[[1]][0, 1], nrow = height, ncol = 0)
combb1 <- function(allsets, number, carry) {
if(number > length(allsets)){
output <<- cbind(output, carry, deparse.level = 0)
return()
} else{
for (j in 1:length(allsets[[number]][1,])) {
# Only add unique combinations (some combinations are vectors)
if((TRUE %in% (allsets[[number]][, j] %in% carry)) == FALSE) {
newcarry <- c(carry, allsets[[number]][, j], use.names = FALSE)
number2 <- number + 1
combb1(allsets, number2, newcarry)
} else{
next()
}
}
}
}
combb1(allsets, 1, carry)
return(output)
}
正如您从该解决方案中看到的,递归很简洁(组合函数)并且不会混淆任何全局/工作区变量。