0

我正在尝试在 R 中编写一个函数，该函数将接受输入numb并输出相应的加泰罗尼亚数字。供您参考，加泰罗尼亚数字的递归公式是，

C_0 = 1; 
C_n = {(4n - 2)*C_(n-1)}/(n+1)

我的代码如下，

catalan_num_recr <- function(numb){
  if (numb == 0){
    return(1)
  }
  else
    return(((4*numb-2)*catalan_num_recr(numb-1))/(numb+1))
}

当我运行该功能时，我得到，

> catalan_num_recr(3)
[1] 5

哪个是对的。

目标：但是，我正在尝试查找加泰罗尼亚语的数字范围，我想找到类似的东西，catalan_num_recr(1:10).

问题：这不适用于我的功能，我收到以下警告，

Warning messages:
1: In if (numb == 0) { :
  the condition has length > 1 and only the first element will be used

还有很多错误的值作为输出，

> catalan_num_recr(1:15)
 [1] 1.000000 2.000000 2.500000 2.800000 3.000000 3.142857 3.250000
 [8] 3.333333 3.400000 3.454545 3.500000 3.538462 3.571429 3.600000
[15] 3.625000

有人可以帮我修改我的功能/或帮我解决问题吗？

问候。

4

5 回答 5

2

使用Vectorize它接受一个函数并返回一个新函数，该函数本质上是接受向量化输入的函数的包装器。计算实际上是重复执行的，mapply因此它不会像编写手动矢量化的函数那样快。

Vectorize(catalan_num_recr)(1:15)
 [1]       1       2       5      14      42     132     429    1430    4862
[10]   16796   58786  208012  742900 2674440 9694845
于 2018-05-21T16:48:00.847 回答
1

您的函数当前不接受向量输入。这可能有点令人困惑，因为许多函数（例如数学运算符）默认情况下是矢量化的。map您需要使用or函数将该函数应用于向量的每个元素apply。这实际上是一个快速循环，并且purrr包具有语法一致和类型稳定的优点。

catalan_num_recr <- function(numb){
  if (numb == 0){
    return(1)
  }
  else
    return(((4*numb-2)*catalan_num_recr(numb-1))/(numb+1))
}

purrr::map_dbl(1:15, catalan_num_recr)
#>  [1]       1       2       5      14      42     132     429    1430
#>  [9]    4862   16796   58786  208012  742900 2674440 9694845

由reprex 包(v0.2.0) 于 2018 年 5 月 21 日创建。

于 2018-05-21T16:48:29.793 回答
1

您只需要使用sapply来自apply家庭的矢量化函数。

sapply(1:10,catalan_num_recr,USE.NAMES = F)

于 2018-05-21T16:49:25.443 回答
1

你可以在你的数字向量上使用sapply（或）lapply

sapply(c(0:15), catalan_num_recr)

只需将输出放入数据框中，以便您检查数字

data.frame("n" = c(0:5), "cnr" = sapply(c(0:5), catalan_num_recr))

  n cnr
1 0   1
2 1   1
3 2   2
4 3   5
5 4  14
6 5  42
于 2018-05-21T16:57:57.147 回答
1

首先，您收到的警告是告诉您在numb == 0R 中只会评估numb[1] == 0.

虽然其他答案是正确的，但它们效率不高，因为您需要多次计算每个加泰罗尼亚数字。（对于 Catalan(15)，您需要计算 Catalan(14)，您已经计算过了）

我认为这是一个更好的版本，它计算每个加泰罗尼亚语数字直到最大数字，然后返回您要求的数字。

catalan <- function(numbs) {
   cat <- vector("numeric", length(max(numbs)) + 1)
   for (i in 0:max(numbs)) {
      if (i == 0) {
         cat[i+1] <- 1
      } else {
         cat[i+1] <- ((4*i - 2)*cat[i])/(i + 1)
      }
   }
   cat[numbs + 1]
}

> catalan(1:15)
 [1]       1       2       5      14      42     132     429
 [8]    1430    4862   16796   58786  208012  742900 2674440
[15] 9694845

这是一个小基准，其中catrvect = Vectorize(catalan_num_recr)

microbenchmark::microbenchmark(catalan(1:100), catvect(1:100))
Unit: microseconds
           expr      min        lq      mean    median       uq
 catalan(1:100)   86.515   98.7565  127.3139  122.7665  141.423
 catvect(1:100) 4764.728 4947.3070 5661.0624 5124.3385 5647.238
       max neval
   276.825   100
 11009.428   100

当然，现在的函数不是递归的。

于 2018-05-21T17:50:22.063 回答